Searched refs:ce_handle (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 448 CollectiveExecutor::Handle* ce_handle = in ExecuteAsync() local
471 delete ce_handle; in ExecuteAsync()
478 handle, step_id, item, rendezvous, ce_handle, collector, cost_graph, in ExecuteAsync()
480 [item, rendezvous, ce_handle, done, start_time_usecs, input_size, in ExecuteAsync()
493 delete ce_handle; in ExecuteAsync()
499 CollectiveExecutor::Handle* ce_handle, StepStatsCollector* collector, in StartParallelExecutors() argument
519 args.collective_executor = ce_handle ? ce_handle->get() : nullptr; in StartParallelExecutors()
|
| D | graph_mgr.h | 168 CollectiveExecutor::Handle* ce_handle,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 634 CollectiveExecutor::Handle ce_handle( in RecvBufAsync() local
636 CollectiveRemoteAccess* rma = ce_handle.get()->remote_access(); in RecvBufAsync()
|