Searched refs:CollectiveContext (Results 1 – 18 of 18) sorted by relevance
33 std::shared_ptr<CollectiveContext> col_ctx) override;42 std::shared_ptr<CollectiveContext> col_ctx_;
62 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
329 auto col_ctx = std::make_shared<CollectiveContext>( in RunReduce()361 auto col_ctx = std::make_shared<CollectiveContext>( in RunBroadcast()402 auto col_ctx = std::make_shared<CollectiveContext>( in RunGather()
59 std::shared_ptr<CollectiveContext> col_ctx) override;67 std::shared_ptr<CollectiveContext> col_ctx_;
43 std::shared_ptr<CollectiveContext> col_ctx) override;84 std::shared_ptr<CollectiveContext> col_ctx_;
43 std::shared_ptr<CollectiveContext> col_ctx) override;112 std::shared_ptr<CollectiveContext> col_ctx_;
54 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
188 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
315 auto col_ctx = std::make_shared<CollectiveContext>( in ExecuteAsync()
245 std::shared_ptr<CollectiveContext> col_ctx) { in InitializeCollectiveContext()
401 auto col_ctx = std::make_shared<CollectiveContext>( in DoPermute()
490 auto col_ctx = std::make_shared<CollectiveContext>( in DoGather()
523 auto col_ctx = std::make_shared<CollectiveContext>( in DoReduce()
684 auto col_ctx = std::make_shared<CollectiveContext>( in DoBroadcast()
364 struct CollectiveContext { struct379 CollectiveContext(CollectiveExecutor* col_exec, argument391 virtual void Enqueue(std::shared_ptr<CollectiveContext> col_ctx,420 std::shared_ptr<CollectiveContext> col_ctx) = 0;
169 CollectiveContext::CollectiveContext( in CollectiveContext() function in tensorflow::CollectiveContext
31 void Enqueue(std::shared_ptr<CollectiveContext> col_ctx,70 void NcclCommunicator::Enqueue(std::shared_ptr<CollectiveContext> col_ctx, in Enqueue()
369 std::shared_ptr<CollectiveContext>) override { in InitializeCollectiveContext() argument