diff options
Diffstat (limited to 'tensorflow/core/framework/tensor_description.proto')
-rw-r--r-- | tensorflow/core/framework/tensor_description.proto | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/core/framework/tensor_description.proto b/tensorflow/core/framework/tensor_description.proto new file mode 100644 index 0000000000..1fff3ee155 --- /dev/null +++ b/tensorflow/core/framework/tensor_description.proto @@ -0,0 +1,19 @@ +syntax = "proto3"; + +package tensorflow; +// option cc_enable_arenas = true; + +import "tensorflow/core/framework/types.proto"; +import "tensorflow/core/framework/tensor_shape.proto"; +import "tensorflow/core/framework/allocation_description.proto"; + +message TensorDescription { + // Data type of tensor elements + DataType dtype = 1; + + // Shape of the tensor. + TensorShapeProto shape = 2; + + // Information about the size and allocator used for the data + AllocationDescription allocation_description = 4; +}; |