Searched defs:group_mode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.cc | 116 CollectiveOpGroupMode group_mode) { in CollectiveOpGroupModeToString()
132 CollectiveOpGroupMode group_mode) { in GetParticipatingDevicesGroups()
241 CollectiveOpGroupMode group_mode) { in GetParticipatingDevices()
|
| D | collective_ops_utils_test.cc | 142 StatusOr<CollectiveOpGroupMode> group_mode = in ToString() local
396 StatusOr<CollectiveOpGroupMode> group_mode = in TEST_P() local
|
| D | collective_decomposer_utils.cc | 35 CollectiveOpGroupMode group_mode, in CreateStartIndicesForCollectiveDecomposition()
|
| D | hlo_verifier.cc | 95 CollectiveOpGroupMode group_mode) { in GetSubgroupSize()
263 CollectiveOpGroupMode group_mode, in CheckReplicaGroups()
623 CollectiveOpGroupMode group_mode) { in CheckDuplicatedSourceOrTarget()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_thunk.cc | 116 CollectiveOpGroupMode group_mode, int64_t op_id) { in LockNcclComm()
|
| D | nccl_collective_thunk.h | 56 CollectiveOpGroupMode group_mode; member
|
| D | jitrt_custom_calls.cc | 345 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 ()()
|
| D | ir_emitter_unnested.cc | 2855 CollectiveOpGroupMode group_mode = NcclThunkType::GetGroupMode(op); in EmitNcclThunk() local
|