| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 31 void Enqueue(std::shared_ptr<CollectiveContext> col_ctx,
70 void NcclCommunicator::Enqueue(std::shared_ptr<CollectiveContext> col_ctx, in Enqueue() argument
72 const CollectiveParams* col_params = col_ctx->col_params; in Enqueue()
77 NcclCollectiveKey(col_ctx->exec_key, col_ctx->step_id); in Enqueue()
78 auto* compute_stream = col_ctx->op_ctx->op_device_context()->stream(); in Enqueue()
79 auto* gpu_info = col_ctx->op_ctx->device()->tensorflow_gpu_device_info(); in Enqueue()
81 compute_stream->parent(), compute_stream, gpu_info, col_ctx->input, in Enqueue()
82 col_ctx->output, col_ctx->col_params->default_rank, in Enqueue()
84 CancellationManager* cancel_mgr = col_ctx->op_ctx->cancellation_manager(); in Enqueue()
115 << col_ctx->device_name << " instance " in Enqueue()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.cc | 62 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext() argument
63 col_ctx_ = col_ctx; in InitializeCollectiveContext()
64 col_params_ = col_ctx->col_params; in InitializeCollectiveContext()
66 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
67 &col_ctx->device_locality); in InitializeCollectiveContext()
|
| D | collective_nccl_test.cc | 329 auto col_ctx = std::make_shared<CollectiveContext>( in RunReduce() local
334 TF_CHECK_OK(reducer->InitializeCollectiveContext(col_ctx)); in RunReduce()
361 auto col_ctx = std::make_shared<CollectiveContext>( in RunBroadcast() local
367 TF_CHECK_OK(broadcaster->InitializeCollectiveContext(col_ctx)); in RunBroadcast()
402 auto col_ctx = std::make_shared<CollectiveContext>( in RunGather() local
408 TF_CHECK_OK(gatherer->InitializeCollectiveContext(col_ctx)); in RunGather()
|
| D | collective_nccl.h | 33 std::shared_ptr<CollectiveContext> col_ctx) override;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | permuter.cc | 54 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext() argument
55 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
56 col_ctx_ = col_ctx; in InitializeCollectiveContext()
57 col_params_ = col_ctx->col_params; in InitializeCollectiveContext()
59 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
60 &col_ctx->device_locality); in InitializeCollectiveContext()
|
| D | hierarchical_tree_broadcaster.cc | 188 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext() argument
189 CHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
190 col_ctx_ = col_ctx; in InitializeCollectiveContext()
191 col_params_ = col_ctx->col_params; in InitializeCollectiveContext()
193 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
194 &col_ctx->device_locality); in InitializeCollectiveContext()
|
| D | ring_alg.cc | 245 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext() argument
246 DCHECK(col_ctx->dev_mgr); in InitializeCollectiveContext()
247 col_ctx_ = col_ctx; in InitializeCollectiveContext()
248 col_params_ = col_ctx->col_params; in InitializeCollectiveContext()
250 col_ctx->dev_mgr, col_ctx->device_name, &col_ctx->device, in InitializeCollectiveContext()
251 &col_ctx->device_locality); in InitializeCollectiveContext()
|
| D | base_collective_executor.cc | 315 auto col_ctx = std::make_shared<CollectiveContext>( in ExecuteAsync() local
318 status = col_impl->InitializeCollectiveContext(col_ctx); in ExecuteAsync()
327 RunClosure([col_impl, col_ctx, done_safe, ctx, in ExecuteAsync()
339 col_impl->Run([col_impl, col_ctx, done_safe](const Status& s) { in ExecuteAsync()
|
| D | permuter.h | 59 std::shared_ptr<CollectiveContext> col_ctx) override;
|
| D | hierarchical_tree_broadcaster.h | 43 std::shared_ptr<CollectiveContext> col_ctx) override;
|
| D | ring_alg.h | 43 std::shared_ptr<CollectiveContext> col_ctx) override;
|
| D | permuter_test.cc | 401 auto col_ctx = std::make_shared<CollectiveContext>( in DoPermute() local
405 TF_CHECK_OK(permuter->InitializeCollectiveContext(col_ctx)); in DoPermute()
|
| D | ring_gatherer_test.cc | 490 auto col_ctx = std::make_shared<CollectiveContext>( in DoGather() local
494 TF_CHECK_OK(gatherer->InitializeCollectiveContext(col_ctx)); in DoGather()
|
| D | ring_reducer_test.cc | 523 auto col_ctx = std::make_shared<CollectiveContext>( in DoReduce() local
527 TF_CHECK_OK(reducer->InitializeCollectiveContext(col_ctx)); in DoReduce()
|
| D | hierarchical_tree_broadcaster_test.cc | 684 auto col_ctx = std::make_shared<CollectiveContext>( in DoBroadcast() local
688 TF_CHECK_OK(broadcaster->InitializeCollectiveContext(col_ctx)); in DoBroadcast()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 391 virtual void Enqueue(std::shared_ptr<CollectiveContext> col_ctx,
420 std::shared_ptr<CollectiveContext> col_ctx) = 0;
|