| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | collective_ops.cc | 588 Status FillCollectiveParams(CollectiveParams* col_params, in FillCollectiveParams()
626 void Run(OpKernelContext* c, CollectiveParams* col_params, in Run()
716 auto col_params = new CollectiveParams(); in ComputeAsync() local
767 auto col_params = new CollectiveParams(); in ComputeAsync() local
813 auto col_params = new CollectiveParams(); in ComputeAsync() local
862 auto col_params = new CollectiveParams(); in ComputeAsync() local
1065 Status FillCollectiveParams(CollectiveParams* col_params, in FillCollectiveParams()
1100 void Run(OpKernelContext* c, CollectiveParams* col_params, in Run()
1182 auto col_params = new CollectiveParams(); in ComputeAsync() local
1232 auto col_params = new CollectiveParams(); in ComputeAsync() local
|
| H A D | collective_nccl.cc | 29 Status NcclBase::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | all_to_all_test.cc | 46 auto col_params = CreateCollectiveParams(*test_env_, i, "AllToAll", in TEST_F() local
78 auto col_params = CreateCollectiveParams( in TEST_F() local
112 auto col_params = CreateCollectiveParams(*test_env_, i, "AllToAll", in TEST_F() local
144 auto col_params = CreateCollectiveParams(*test_env_, i, "AllToAll", in TEST_F() local
|
| H A D | base_collective_executor.cc | 273 const CollectiveParams* col_params, in ExecuteAsync()
406 const CollectiveParams& col_params, in CreateCollective()
457 const CollectiveParams& col_params) { in CheckDependencies()
470 const CollectiveParams& col_params) { in WaitForDependencies()
479 const CollectiveParams& col_params) { in UnblockDependencies()
|
| H A D | ring_alg.cc | 110 Status GenerateSubdivsInCollectiveParams(CollectiveParams* col_params) { in GenerateSubdivsInCollectiveParams()
180 Status RingAlg::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| H A D | collective_test_util.cc | 190 core::RefCountPtr<CollectiveParams> col_params(new CollectiveParams()); in CreateCollectiveParams() local
291 Status RunCollective(CollectiveTestEnv* test_env, CollectiveParams* col_params, in RunCollective()
|
| H A D | permuter.h | 52 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams()
|
| H A D | all_to_all.h | 36 Status InitializeCollectiveParams(CollectiveParams* col_params) override { in InitializeCollectiveParams()
|
| H A D | collective_util.cc | 54 string SubdivPermDebugString(const CollectiveParams& col_params) { in SubdivPermDebugString()
|
| H A D | ring_gatherer.cc | 45 Status RingGatherer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| H A D | ring_reducer.cc | 48 Status RingReducer::InitializeCollectiveParams(CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| H A D | hierarchical_tree_broadcaster.cc | 79 CollectiveParams* col_params) { in InitializeCollectiveParams()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | collective.h | 316 const CollectiveParams* col_params, in ExecuteAsync()
358 virtual void WaitForDependencies(const CollectiveParams& col_params) {} in WaitForDependencies()
362 virtual void UnblockDependencies(const CollectiveParams& col_params) {} in UnblockDependencies()
397 core::IntrusivePtr<const CollectiveParams> col_params; member
|
| H A D | collective.cc | 158 OpKernelContext::Params* op_params, const CollectiveParams* col_params, in CollectiveContext()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | collective_communicator.cc | 83 const CollectiveParams* col_params = col_ctx->col_params.get(); in Enqueue() local
|