xref: /aosp_15_r20/external/tensorflow/tensorflow/python/framework/cpp_shape_inference.proto (revision b6fb3261f9314811a0f4371741dbb8839866f948) 
1syntax = "proto3";
2
3package tensorflow;
4
5import "tensorflow/core/framework/full_type.proto";
6import "tensorflow/core/framework/tensor_shape.proto";
7import "tensorflow/core/framework/types.proto";
8
9option cc_enable_arenas = true;
10option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/python/framework/cpp_shape_inference_go_proto";
11
12message CppShapeInferenceResult {
13  message HandleShapeAndType {
14    reserved 3;
15
16    TensorShapeProto shape = 1;
17    DataType dtype = 2;
18    FullTypeDef type = 4;
19  }
20  message HandleData {
21    bool is_set = 1;
22
23    // Only valid if <is_set>.
24    repeated HandleShapeAndType shape_and_type = 2;
25  }
26  TensorShapeProto shape = 1;
27
28  reserved 2;  // was handle_shape
29  reserved 3;  // was handle_dtype
30  HandleData handle_data = 4;
31}
32
33message CppShapeInferenceInputsNeeded {
34  repeated int32 input_tensors_needed = 1;
35  repeated int32 input_tensors_as_shapes_needed = 2;
36}
37