1syntax = "proto3"; 2 3package tensorflow.eager; 4 5import "tensorflow/core/framework/tensor_shape.proto"; 6import "tensorflow/core/framework/types.proto"; 7 8option cc_enable_arenas = true; 9option java_outer_classname = "RemoteTensorHandleProtos"; 10option java_multiple_files = true; 11option java_package = "org.tensorflow.framework"; 12option go_package = "github.com/tensorflow/tensorflow/tensorflow/go/core/protobuf/for_core_protos_go_proto"; 13 14message ResourceDtypeAndShape { 15 DataType dtype = 1; 16 TensorShapeProto shape = 2; 17} 18 19message RemoteTensorHandle { 20 // The ID of the operation that produced this tensor. 21 int64 op_id = 1; 22 // The index into the outputs of the operation that produced this tensor. 23 int32 output_num = 2; 24 // Device where the tensor is located. Cannot be empty. 25 // For multi-device functions, it's the default device passed to placer. 26 string device = 3; 27 // Device of the operation producing this tensor. Can be empty if the 28 // operation producing this tensor is a multi-device function. 29 string op_device = 4; 30 // Tensor type. 31 DataType dtype = 5; 32 // Optional data types and shapes of a remote resource variable. 33 repeated ResourceDtypeAndShape resource_dtypes_and_shapes = 6; 34} 35