diff options
Diffstat (limited to 'tensorflow/compiler/aot/tfcompile.proto')
-rw-r--r-- | tensorflow/compiler/aot/tfcompile.proto | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/tensorflow/compiler/aot/tfcompile.proto b/tensorflow/compiler/aot/tfcompile.proto new file mode 100644 index 0000000000..be3f504350 --- /dev/null +++ b/tensorflow/compiler/aot/tfcompile.proto @@ -0,0 +1,43 @@ +syntax = "proto3"; + +package tensorflow.tfcompile; +option cc_enable_arenas = true; +option java_outer_classname = "CompileProtos"; +option java_multiple_files = true; +option java_package = "org.tensorflow.tfcompile"; + +import "tensorflow/core/framework/tensor_shape.proto"; + +// TensorId identifies a tensor in a TensorFlow graph, by specifying the output +// index of a particular node in the graph. If the output of the named node +// feeds into other node(s), this corresponds to one or more edges. Otherwise +// it doesn't correspond to any existing edges at all, e.g. for output nodes. +message TensorId { + string node_name = 1; + int64 output_index = 2; +}; + +// Feed represents a single feed tensor in the graph, which corresponds to an +// input argument for the generated function. +message Feed { + TensorId id = 1; + TensorShapeProto shape = 2; + string name = 3; // Optional name for generated code. +}; + +// Fetch represents a single fetch tensor in the graph, which corresponds to an +// output argument for the generated function. +message Fetch { + TensorId id = 1; + string name = 2; // Optional name for generated code. +}; + +// Config represents configuration information for tfcompile. +message Config { + // Each feed is a positional input argument for the generated function. The + // order of each entry matches the order of each input argument. + repeated Feed feed = 1; + // Each fetch is a positional output argument for the generated function. The + // order of each entry matches the order of each output argument. + repeated Fetch fetch = 2; +}; |