Searched refs:col_impl (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 246 CollectiveImplementationInterface* col_impl = nullptr; in ExecuteAsync() local
247 Status status = CreateCollective(col_params, &col_impl); in ExecuteAsync()
250 DCHECK_EQ(nullptr, col_impl); in ExecuteAsync()
256 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
260 delete col_impl; in ExecuteAsync()
267 SchedClosure([col_impl, col_ctx, done_safe]() { in ExecuteAsync()
268 col_impl->Run([col_impl, col_ctx, done_safe](const Status& s) { in ExecuteAsync()
271 delete col_impl; in ExecuteAsync()
285 CollectiveImplementationInterface** col_impl) { in CreateCollective() argument
286 *col_impl = nullptr; in CreateCollective()
[all …]
|
| D | collective_param_resolver_local.cc | 671 CollectiveImplementationInterface* col_impl; in CompleteInstanceFromInitializedIRec() local
673 cp->instance.impl_details.collective_name, &col_impl); in CompleteInstanceFromInitializedIRec()
675 status = col_impl->InitializeInstanceBeforeGroupDiscovery(cp); in CompleteInstanceFromInitializedIRec()
692 [col_impl, ir, device, cp, done](InstanceRec* irec) { in CompleteInstanceFromInitializedIRec()
705 s = col_impl->InitializeCollectiveParams(cp); in CompleteInstanceFromInitializedIRec()
710 done(col_impl->InitializeCollectiveParams(cp)); in CompleteInstanceFromInitializedIRec()
|
| D | base_collective_executor.h | 167 CollectiveImplementationInterface** col_impl);
|