Searched refs:BROADCAST_COLLECTIVE (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster_test.cc | 221 BROADCAST_COLLECTIVE, dtype, shape); in DeviceInstance()
286 BROADCAST_COLLECTIVE, DT_FLOAT, TensorShape({1})); in TEST_F()
312 BROADCAST_COLLECTIVE, DT_FLOAT, TensorShape({1})); in TEST_F()
356 cp->instance.type = BROADCAST_COLLECTIVE; in TEST_F()
|
| D | collective_util.cc | 73 if (col_params.instance.type == BROADCAST_COLLECTIVE) { in SubdivPermDebugString()
|
| D | base_collective_executor.cc | 310 (col_params->instance.type == BROADCAST_COLLECTIVE && in ExecuteAsync()
414 if (col_params.instance.type == BROADCAST_COLLECTIVE) { in CreateCollective()
|
| D | collective_param_resolver_local_test.cc | 219 cp->instance.type = BROADCAST_COLLECTIVE; in InitializeCollectiveParamsForBroadcast()
313 cp->instance.type = BROADCAST_COLLECTIVE; in MakeCollectiveParams()
|
| D | collective_param_resolver_local.cc | 66 case BROADCAST_COLLECTIVE: in GetCollectiveName()
781 if (cp->instance.type == BROADCAST_COLLECTIVE) { in CompleteInstanceFromInitializedIRec()
|
| D | hierarchical_tree_broadcaster.cc | 80 CHECK_EQ(col_params->instance.type, BROADCAST_COLLECTIVE); in InitializeCollectiveParams()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_broadcaster.h | 25 NcclBroadcaster() : NcclBase(BROADCAST_COLLECTIVE, "NcclBroadcast") {} in NcclBroadcaster()
|
| D | collective_nccl.cc | 41 case BROADCAST_COLLECTIVE: in InitializeCollectiveParams()
|
| D | collective_ops.cc | 339 col_params_->instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastSendOpKernel()
416 col_params_->instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastRecvOpKernel()
819 FillCollectiveParams(col_params, BROADCAST_COLLECTIVE, in ComputeAsync()
868 FillCollectiveParams(col_params, BROADCAST_COLLECTIVE, in ComputeAsync()
|
| D | collective_nccl_test.cc | 487 : NcclTestBase(/*collective_type=*/BROADCAST_COLLECTIVE, in NcclBroadcasterTest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed_test.cc | 358 DefineCollectiveParams(num_workers, num_devices, "CPU", BROADCAST_COLLECTIVE, in TEST_F()
369 DefineCollectiveParams(num_workers, num_devices, "CPU", BROADCAST_COLLECTIVE, in TEST_F()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | collective_communicator.cc | 155 case BROADCAST_COLLECTIVE: { in Enqueue()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 47 BROADCAST_COLLECTIVE, enumerator
|