Home
last modified time | relevance | path

Searched refs:BROADCAST_COLLECTIVE (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc72 if (col_params.instance.type == BROADCAST_COLLECTIVE) { in SubdivPermDebugString()
Dhierarchical_tree_broadcaster_test.cc264 col_params_.instance.type = BROADCAST_COLLECTIVE; in Init()
539 cp->instance.type = BROADCAST_COLLECTIVE; in PrepColParamsForSubdivPermsTest()
791 cp.instance.type = BROADCAST_COLLECTIVE; in TEST_F()
Dbase_collective_executor.cc242 (col_params.instance.type == BROADCAST_COLLECTIVE && in ExecuteAsync()
Dcollective_param_resolver_local_test.cc211 cp->instance.type = BROADCAST_COLLECTIVE; in InitializeCollectiveParamsForBroadcast()
Dcollective_param_resolver_local.cc613 if (cp->instance.type == BROADCAST_COLLECTIVE) { in AssignCollectiveType()
686 bool broadcast = cp->instance.type == BROADCAST_COLLECTIVE; in CompleteInstanceFromInitializedIRec()
Dhierarchical_tree_broadcaster.cc78 CHECK_EQ(col_params->instance.type, BROADCAST_COLLECTIVE); in InitializeCollectiveParams()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc235 col_params_.instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastSendOpKernel()
304 col_params_.instance.type = BROADCAST_COLLECTIVE; in CollectiveBcastRecvOpKernel()
/external/tensorflow/tensorflow/core/framework/
Dcollective.h44 BROADCAST_COLLECTIVE, enumerator