/external/tensorflow/tensorflow/core/common_runtime/ |
collective_executor_mgr.cc | 59 CollectiveRemoteAccessLocal* rma = local 61 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_,
|
/external/tensorflow/tensorflow/contrib/gdr/ |
gdr_collective_executor_mgr.cc | 151 CollectiveRemoteAccessDistributed* rma = local 155 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_,
|
gdr_worker.cc | 161 CollectiveRemoteAccess* rma = ce_handle.get()->remote_access(); local 162 rma->buf_rendezvous()->ConsumeBuf(
|
/external/tensorflow/tensorflow/core/distributed_runtime/ |
rpc_collective_executor_mgr.cc | 49 CollectiveRemoteAccessDistributed* rma = local 52 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_,
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
grpc_worker_service.cc | 566 CollectiveRemoteAccess* rma = ce_handle.get()->remote_access(); local 567 rma->buf_rendezvous()->ConsumeBuf(
|