Searched refs:col_impl (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | base_collective_executor.cc | 307 CollectiveImplementationInterface* col_impl = nullptr; in ExecuteAsync() local
308 Status status = CreateCollective(*col_params, &col_impl); in ExecuteAsync()
311 DCHECK_EQ(nullptr, col_impl); in ExecuteAsync()
314 core::ScopedUnref unref(col_impl); in ExecuteAsync()
318 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
325 col_impl->Ref(); in ExecuteAsync()
327 RunClosure([col_impl, col_ctx, done_safe, ctx, in ExecuteAsync()
329 core::ScopedUnref unref(col_impl); in ExecuteAsync()
338 col_impl->Ref(); in ExecuteAsync()
339 col_impl->Run([col_impl, col_ctx, done_safe](const Status& s) { in ExecuteAsync()
[all …]
|
| D | collective_param_resolver_local.cc | 146 CollectiveImplementationInterface* col_impl; in CompleteGroupLocal() local
151 &col_impl); in CompleteGroupLocal()
155 GetCollectiveName(cp, /*nccl=*/false), &col_impl); in CompleteGroupLocal()
158 status = col_impl->InitializeCollectiveGroupRuntimeDetails( in CompleteGroupLocal()
675 CollectiveImplementationInterface* col_impl; in AssignCollectiveType() local
678 CollectiveRegistry::LookupParamResolverInstance("NcclReduce", &col_impl) in AssignCollectiveType()
752 CollectiveImplementationInterface* col_impl; in CompleteInstanceFromInitializedIRec() local
754 cp->instance.impl_details.collective_name, &col_impl); in CompleteInstanceFromInitializedIRec()
764 [col_impl, ir, device, cp, done](InstanceRec* irec) { in CompleteInstanceFromInitializedIRec()
775 s = col_impl->InitializeCollectiveParams(cp); in CompleteInstanceFromInitializedIRec()
[all …]
|
| D | base_collective_executor.h | 156 CollectiveImplementationInterface** col_impl);
|