Searched refs:col_impl (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 249 CollectiveImplementationInterface* col_impl = nullptr; in ExecuteAsync() local
250 Status status = CreateCollective(col_params, &col_impl); in ExecuteAsync()
253 DCHECK_EQ(nullptr, col_impl); in ExecuteAsync()
259 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
263 delete col_impl; in ExecuteAsync()
268 remote_access_->RunClosure([col_impl, col_ctx, done_safe, ctx]() { in ExecuteAsync()
276 col_impl->Run([col_impl, col_ctx, done_safe](const Status& s) { in ExecuteAsync()
279 delete col_impl; in ExecuteAsync()
293 CollectiveImplementationInterface** col_impl) { in CreateCollective() argument
297 *col_impl = nullptr; in CreateCollective()
[all …]
|
| D | collective_param_resolver_local.cc | 94 CollectiveImplementationInterface* col_impl; in CompleteGroupLocal() local
99 &col_impl); in CompleteGroupLocal()
103 GetCollectiveName(cp, /*nccl=*/false), &col_impl); in CompleteGroupLocal()
106 status = col_impl->InitializeCollectiveGroupRuntimeDetails( in CompleteGroupLocal()
692 CollectiveImplementationInterface* col_impl; in AssignCollectiveType() local
695 CollectiveRegistry::LookupParamResolverInstance("NcclReduce", &col_impl) in AssignCollectiveType()
754 CollectiveImplementationInterface* col_impl; in CompleteInstanceFromInitializedIRec() local
756 cp->instance.impl_details.collective_name, &col_impl); in CompleteInstanceFromInitializedIRec()
766 [col_impl, ir, device, cp, done](InstanceRec* irec) { in CompleteInstanceFromInitializedIRec()
778 s = col_impl->InitializeCollectiveParams(cp); in CompleteInstanceFromInitializedIRec()
[all …]
|
| D | base_collective_executor.h | 172 CollectiveImplementationInterface** col_impl);
|