8310eda05e
Only used for TensorList at the moment, but we'll likely need to add similar information for other variants as they'll also need special cases in tf.vectorized_map. PiperOrigin-RevId: 334708004 Change-Id: I484845f855c5c9cfeea13b78a520a1d7c60c9fc5
36 lines
1.0 KiB
Protocol Buffer
36 lines
1.0 KiB
Protocol Buffer
syntax = "proto3";
|
|
|
|
package tensorflow;
|
|
option cc_enable_arenas = true;
|
|
option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/python/framework/cpp_shape_inference_go_proto";
|
|
|
|
import "tensorflow/core/framework/types.proto";
|
|
import "tensorflow/core/framework/tensor_shape.proto";
|
|
|
|
message CppShapeInferenceResult {
|
|
message HandleShapeAndType {
|
|
TensorShapeProto shape = 1;
|
|
DataType dtype = 2;
|
|
// For dtype==DT_VARIANT, specialized_type may indicate a more specific
|
|
// type. For other dtypes or when the information is unavailable it is set
|
|
// to ST_INVALID.
|
|
SpecializedType specialized_type = 3;
|
|
}
|
|
message HandleData {
|
|
bool is_set = 1;
|
|
|
|
// Only valid if <is_set>.
|
|
repeated HandleShapeAndType shape_and_type = 2;
|
|
}
|
|
TensorShapeProto shape = 1;
|
|
|
|
reserved 2; // was handle_shape
|
|
reserved 3; // was handle_dtype
|
|
HandleData handle_data = 4;
|
|
}
|
|
|
|
message CppShapeInferenceInputsNeeded {
|
|
repeated int32 input_tensors_needed = 1;
|
|
repeated int32 input_tensors_as_shapes_needed = 2;
|
|
}
|