• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1syntax = "proto3";
2
3package tensorflow;
4option cc_enable_arenas = true;
5option java_outer_classname = "TensorDescriptionProtos";
6option java_multiple_files = true;
7option java_package = "org.tensorflow.framework";
8option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/framework";
9import "tensorflow/core/framework/types.proto";
10import "tensorflow/core/framework/tensor_shape.proto";
11import "tensorflow/core/framework/allocation_description.proto";
12
13message TensorDescription {
14  // Data type of tensor elements
15  DataType dtype = 1;
16
17  // Shape of the tensor.
18  TensorShapeProto shape = 2;
19
20  // Information about the size and allocator used for the data
21  AllocationDescription allocation_description = 4;
22};
23