internal

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

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

Functions

func GenerateFunctionsForRegisteredOps

func GenerateFunctionsForRegisteredOps(w io.Writer, apidefs []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.