Home
last modified time | relevance | path

Searched defs:col_params (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbase_collective_executor.cc219 const CollectiveParams& col_params, in ExecuteAsync()
284 const CollectiveParams& col_params, in CreateCollective()
312 const CollectiveParams& col_params) { in CheckDependencies()
325 const CollectiveParams& col_params) { in WaitForDependencies()
333 void BaseCollectiveExecutor::Launched(const CollectiveParams& col_params) { in Launched()
Dring_alg.cc108 Status GenerateSubdivsInCollectiveParams(CollectiveParams* col_params) { in GenerateSubdivsInCollectiveParams()
164 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
Dcollective_util.cc54 string SubdivPermDebugString(const CollectiveParams& col_params) { in SubdivPermDebugString()
Dring_gatherer.cc43 Status RingGatherer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
Dring_reducer.cc46 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
Dhierarchical_tree_broadcaster.cc77 CollectiveParams* col_params) { in InitializeCollectiveParams()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl_reducer.cc31 Status NcclReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
53 CollectiveParams* col_params) { in InitializeInstanceBeforeGroupDiscovery()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h262 const CollectiveParams& col_params, in ExecuteAsync()
288 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()
292 virtual void Launched(const CollectiveParams& col_params) {} in Launched()
352 const CollectiveParams& col_params; variable
Dcollective.cc160 const CollectiveParams& col_params, in CollectiveContext()