Home
last modified time | relevance | path

Searched defs:col_ctx (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcollective_nccl.cc62 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
Dcollective_nccl_test.cc331 auto col_ctx = std::make_shared<CollectiveContext>( in RunReduce() local
363 auto col_ctx = std::make_shared<CollectiveContext>( in RunBroadcast() local
404 auto col_ctx = std::make_shared<CollectiveContext>( in RunGather() local
/external/tensorflow/tensorflow/core/common_runtime/
Dpermuter.cc54 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
Dall_to_all.cc79 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
Dring_alg.cc259 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
Dhierarchical_tree_broadcaster.cc189 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
Dcollective_test_util.cc352 auto col_ctx = std::make_shared<CollectiveContext>( in RunCollective() local
Dbase_collective_executor.cc322 auto col_ctx = std::make_shared<CollectiveContext>( in ExecuteAsync() local
/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc81 void NcclCommunicator::Enqueue(std::shared_ptr<CollectiveContext> col_ctx, in Enqueue()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_param_resolver_distributed_test.cc99 void Enqueue(std::shared_ptr<CollectiveContext> col_ctx, in Enqueue()