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