1syntax = "proto3"; 2 3package tensorflow.tf2xla; 4option cc_enable_arenas = true; 5option java_outer_classname = "Tf2XlaProtos"; 6option java_multiple_files = true; 7option java_package = "org.tensorflow.tf2xla"; 8 9import "tensorflow/core/framework/tensor_shape.proto"; 10import "tensorflow/core/framework/types.proto"; 11 12// TensorMetadata indicates the type and shape of a Tensor that is 13// part of a host compute transfer. 14message TensorMetadata { 15 DataType type = 1; 16 TensorShapeProto shape = 2; 17} 18 19// HostTransferMetadata describes a transfer either from host to device 20// or device to host. It has a key that is unique to the computation, 21// and metadata about the list of tensors being transferred. 22message HostTransferMetadata { 23 // The key used to identify this transfer. 24 string key = 1; 25 26 // For each Tensor being transferred, its type and shape. 27 repeated TensorMetadata metadata = 2; 28} 29 30// HostComputeMetadata describes all the sends and recvs 31// from all host compute transfer ops in a computation. 32message HostComputeMetadata { 33 // Metadata about each device_to_host transfer 34 repeated HostTransferMetadata device_to_host = 1; 35 36 // Metadata about each host_to_device transfer 37 repeated HostTransferMetadata host_to_device = 2; 38} 39