diff options
Diffstat (limited to 'tensorflow/contrib/mpi/mpi_msg.proto')
-rw-r--r-- | tensorflow/contrib/mpi/mpi_msg.proto | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/tensorflow/contrib/mpi/mpi_msg.proto b/tensorflow/contrib/mpi/mpi_msg.proto new file mode 100644 index 0000000000..36f1504901 --- /dev/null +++ b/tensorflow/contrib/mpi/mpi_msg.proto @@ -0,0 +1,19 @@ + +syntax = "proto3"; + +package tensorflow; +option cc_enable_arenas = true; + +import "tensorflow/core/protobuf/worker.proto"; + + +message MPIRecvTensorResponse { + RecvTensorResponse response = 1; + bool singleSend = 2; + string key = 3; + int64 step_id = 4; + uint64 checksum = 5; +} + + + |