internal

package
v2.16.1+incompatible Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 6, 2024 License: Apache-2.0, BSD-2-Clause Imports: 16 Imported by: 0

Documentation

Overview

Package internal generates Go source code with functions for TensorFlow operations.

The basic outline of the generated API is as follows:

- One function for each TensorFlow operation - The arguments to the function are the inputs and required attributes of the operation - The function returns the outputs - A function is also generated for each optional attribute of the operation.

There is a possibility that there are name collisions between the functions generated for ops and the functions generated for optional attributes. For now, we ignore those, but will need to revisit if a collision is actually encountered.

Package internal generates Go source code with functions for TensorFlow operations.

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func GenerateFunctionsForRegisteredOps

func GenerateFunctionsForRegisteredOps(
	w io.Writer, apidefDirs []string) error

GenerateFunctionsForRegisteredOps writes a Go source code file to w containing functions for each TensorFlow operation registered in the address space of the calling process. apidefDirs should be a contain of directories containing api_def_*.pbtxt files to load.

Types

This section is empty.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL