Home
last modified time | relevance | path

Searched defs:group_mode (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.cc116 CollectiveOpGroupMode group_mode) { in CollectiveOpGroupModeToString()
132 CollectiveOpGroupMode group_mode) { in GetParticipatingDevicesGroups()
241 CollectiveOpGroupMode group_mode) { in GetParticipatingDevices()
Dcollective_ops_utils_test.cc142 StatusOr<CollectiveOpGroupMode> group_mode = in ToString() local
396 StatusOr<CollectiveOpGroupMode> group_mode = in TEST_P() local
Dcollective_decomposer_utils.cc35 CollectiveOpGroupMode group_mode, in CreateStartIndicesForCollectiveDecomposition()
Dhlo_verifier.cc95 CollectiveOpGroupMode group_mode) { in GetSubgroupSize()
263 CollectiveOpGroupMode group_mode, in CheckReplicaGroups()
623 CollectiveOpGroupMode group_mode) { in CheckDuplicatedSourceOrTarget()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnccl_collective_thunk.cc116 CollectiveOpGroupMode group_mode, int64_t op_id) { in LockNcclComm()
Dnccl_collective_thunk.h56 CollectiveOpGroupMode group_mode; member
Djitrt_custom_calls.cc345 int64_t group_mode, int64_t op_id, in GetNcclComm()
1608 int64_t group_mode, int64_t op_id, in operator ()()
1676 int64_t group_mode, int64_t op_id, in operator ()()
1805 int64_t group_mode, int64_t op_id, in operator ()()
1873 int64_t group_mode, int64_t op_id, in operator ()()
1937 int64_t group_mode, bool has_split_dimension, in operator ()()
2005 int32_t uid, int64_t group_mode, int64_t op_id, in operator ()()
Dir_emitter_unnested.cc2855 CollectiveOpGroupMode group_mode = NcclThunkType::GetGroupMode(op); in EmitNcclThunk() local