1syntax = "proto3"; 2 3package tensorflow; 4option cc_enable_arenas = true; 5 6import "tensorflow/core/framework/types.proto"; 7import "tensorflow/core/framework/tensor_shape.proto"; 8 9message CppShapeInferenceResult { 10 message HandleShapeAndType { 11 TensorShapeProto shape = 1; 12 DataType dtype = 2; 13 } 14 message HandleData { 15 bool is_set = 1; 16 17 // Only valid if <is_set>. 18 repeated HandleShapeAndType shape_and_type = 2; 19 } 20 TensorShapeProto shape = 1; 21 22 reserved 2; // was handle_shape 23 reserved 3; // was handle_dtype 24 HandleData handle_data = 4; 25} 26 27message CppShapeInferenceInputsNeeded { 28 repeated int32 input_tensors_needed = 1; 29 repeated int32 input_tensors_as_shapes_needed = 2; 30} 31