Searched refs:CollectiveType (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl.h | 25 explicit NcclBase(CollectiveType type, const string& name);
40 const CollectiveType type_;
|
| D | collective_nccl.cc | 26 NcclBase::NcclBase(CollectiveType type, const string& name) in NcclBase()
|
| D | collective_nccl_test.cc | 85 NcclTestBase(CollectiveType collective_type, const string& collective_name) in NcclTestBase()
432 CollectiveType collective_type_;
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_alg.h | 33 explicit RingAlg(CollectiveType type, const string& name);
110 const CollectiveType type_;
|
| D | ring_alg.cc | 99 RingAlg::RingAlg(CollectiveType type, const string& name) in RingAlg()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.h | 203 enum CollectiveType { enum
227 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.cc | 36 const CollectiveType& collective_type, in CompleteGroupCall()
145 cp->instance.type = CollectiveType(request->collective_type()); in CompleteGroupAsync()
191 cp->instance.type = CollectiveType(request->type()); in CompleteInstanceAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 43 enum CollectiveType { enum
109 CollectiveType type = UNDEFINED_COLLECTIVE;
|