Home
last modified time | relevance | path

Searched defs:grouped_sharding (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc1341 GroupedSharding grouped_sharding, const GroupedSharding& reference, in AlignGroupsWithInternal()
1415 GroupedSharding AlignGroupsWith(GroupedSharding grouped_sharding, in AlignGroupsWith()
1424 GroupedSharding grouped_sharding, const GroupedSharding& reference) { in AlignGroupsWithIfCompatible()
1442 Shape GetPerGroupBaseShape(const GroupedSharding& grouped_sharding, in GetPerGroupBaseShape()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util.cc1729 HloSharding UngroupSharding(const GroupedSharding& grouped_sharding) { in UngroupSharding()