Searched refs:collective_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 36 const CollectiveType& collective_type, in CompleteGroupCall() argument
44 req_.set_collective_type(collective_type); in CompleteGroupCall()
145 cp->instance.type = CollectiveType(request->collective_type()); in CompleteGroupAsync()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 491 CollectiveType collective_type, in AddParticipant() argument
509 context.collective_key, data_type, collective_type, reduction_op, in AddParticipant()
536 if (collective->status.ok() && collective->type != collective_type) { in AddParticipant()
539 collective->type, " but now got type ", collective_type); in AddParticipant()
|
| D | nccl_manager.h | 227 const Context& context, CollectiveType collective_type,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 85 NcclTestBase(CollectiveType collective_type, const string& collective_name) in NcclTestBase() argument
86 : collective_type_(collective_type), in NcclTestBase()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 548 int32 collective_type = 5; field
|