Home
last modified time | relevance | path

Searched refs:CollectiveOpGroupModeToString (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcollective_ops_utils_test.cc145 s << CollectiveOpGroupModeToString(*group_mode); in ToString()
H A Dcollective_ops_utils.h93 absl::string_view CollectiveOpGroupModeToString(
H A Dcollective_ops_utils.cc115 absl::string_view CollectiveOpGroupModeToString( in CollectiveOpGroupModeToString() function
H A Dhlo_verifier.cc300 << "In " << CollectiveOpGroupModeToString(group_mode) in CheckReplicaGroups()
307 << "In " << CollectiveOpGroupModeToString(group_mode) in CheckReplicaGroups()
315 << "In " << CollectiveOpGroupModeToString(group_mode) in CheckReplicaGroups()
H A Dwhile_loop_all_reduce_code_motion.cc87 << CollectiveOpGroupModeToString(all_reduce_group_mode); in IsValueReplicatedWithinEachAllReduceGroup()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
H A Dlmhlo_to_jitrt.cc1331 << CollectiveOpGroupModeToString(config.group_mode) in matchAndRewrite()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emitter_unnested.cc2862 CollectiveOpGroupModeToString(group_mode), op.getOperands().size(), in EmitNcclThunk()