Searched refs:BaseCollectiveExecutor (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 211 BaseCollectiveExecutor::~BaseCollectiveExecutor() {} in ~BaseCollectiveExecutor()
213 void BaseCollectiveExecutor::StartAbort(const Status& s) { in StartAbort()
218 void BaseCollectiveExecutor::ExecuteAsync(OpKernelContext* ctx, in ExecuteAsync()
276 void BaseCollectiveExecutor::CompleteParamsAsync( in CompleteParamsAsync()
283 Status BaseCollectiveExecutor::CreateCollective( in CreateCollective()
311 bool BaseCollectiveExecutor::CheckDependencies( in CheckDependencies()
324 void BaseCollectiveExecutor::WaitForDependencies( in WaitForDependencies()
333 void BaseCollectiveExecutor::Launched(const CollectiveParams& col_params) { in Launched()
|
| D | base_collective_executor.h | 94 class BaseCollectiveExecutor : public CollectiveExecutor {
96 BaseCollectiveExecutor(CollectiveExecutorMgrInterface* cem, in BaseCollectiveExecutor() function
106 ~BaseCollectiveExecutor() override;
|
| D | collective_executor_mgr.cc | 61 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| D | ring_gatherer_test.cc | 173 col_exec_ = new BaseCollectiveExecutor( in Init()
|
| D | ring_reducer_test.cc | 195 col_exec_ = new BaseCollectiveExecutor( in Init()
|
| D | hierarchical_tree_broadcaster_test.cc | 253 col_exec_ = new BaseCollectiveExecutor( in Init()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 155 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpc_collective_executor_mgr.cc | 52 return new BaseCollectiveExecutor(this, rma, step_id, dev_mgr_, in Create()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_reducer_test.cc | 111 col_exec_ = new BaseCollectiveExecutor( in Init()
|