tensorflow: github.com/tensorflow/tensorflow/tensorflow/go/genop/internal Index | Files

package internal

import "github.com/tensorflow/tensorflow/tensorflow/go/genop/internal"

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

Package Files

api_def_map.go genop.go lib.go

func GenerateFunctionsForRegisteredOps Uses

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.

Package internal imports 13 packages (graph) and is imported by 1 packages. Updated 2019-01-14. Refresh now. Tools for package owners.