Searched refs:CollectiveType (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.h | 25 explicit NcclBase(CollectiveType type, const string& name);
36 const CollectiveType type_;
|
| D | collective_nccl.cc | 26 NcclBase::NcclBase(CollectiveType type, const string& name) in NcclBase()
|
| D | collective_nccl_test.cc | 86 NcclTestBase(CollectiveType collective_type, const string& collective_name) in NcclTestBase()
434 CollectiveType collective_type_;
|
| D | collective_ops.cc | 589 CollectiveType collective_type, in FillCollectiveParams()
1067 CollectiveType collective_type, in FillCollectiveParams()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.h | 33 explicit RingAlg(CollectiveType type, const string& name);
104 const CollectiveType type_;
|
| D | collective_test_util.h | 94 CollectiveType collective_type, DataType dtype, const TensorShape& shape,
|
| D | collective_test_util.cc | 186 CollectiveType collective_type, DataType dtype, const TensorShape& shape, in CreateCollectiveParams()
|
| D | ring_alg.cc | 100 RingAlg::RingAlg(CollectiveType type, const string& name) in RingAlg()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 204 enum CollectiveType { enum
228 const Context& context, CollectiveType collective_type,
|
| D | nccl_manager.cc | 164 CollectiveType type_in, ncclRedOp_t reduction_op_in, in Collective()
190 const CollectiveType type;
491 CollectiveType collective_type, in AddParticipant()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 165 CollectiveType coll_type = REDUCTION_COLLECTIVE, in DefineCollectiveParams()
182 CollectiveType coll_type, in CreateCollectiveParams()
284 CollectiveType coll_type = REDUCTION_COLLECTIVE, in RestartWorker()
|
| D | collective_param_resolver_distributed.cc | 173 cp->instance.type = CollectiveType(request->type()); in CompleteInstanceAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 45 enum CollectiveType { enum
125 CollectiveType type = UNDEFINED_COLLECTIVE;
|