1syntax = "proto3"; 2 3package tensorflow; 4option cc_enable_arenas = true; 5option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/python/framework/cpp_shape_inference_go_proto"; 6 7import "tensorflow/core/framework/types.proto"; 8import "tensorflow/core/framework/tensor_shape.proto"; 9 10message CppShapeInferenceResult { 11 message HandleShapeAndType { 12 TensorShapeProto shape = 1; 13 DataType dtype = 2; 14 // For dtype==DT_VARIANT, specialized_type may indicate a more specific 15 // type. For other dtypes or when the information is unavailable it is set 16 // to ST_INVALID. 17 SpecializedType specialized_type = 3; 18 } 19 message HandleData { 20 bool is_set = 1; 21 22 // Only valid if <is_set>. 23 repeated HandleShapeAndType shape_and_type = 2; 24 } 25 TensorShapeProto shape = 1; 26 27 reserved 2; // was handle_shape 28 reserved 3; // was handle_dtype 29 HandleData handle_data = 4; 30} 31 32message CppShapeInferenceInputsNeeded { 33 repeated int32 input_tensors_needed = 1; 34 repeated int32 input_tensors_as_shapes_needed = 2; 35} 36