Home
last modified time | relevance | path

Searched refs:SubgroupManualDim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.h337 int64_t SubgroupManualDim() const { in SubgroupManualDim() function
Dhlo_sharding_util.cc139 int64_t to_merge_man_dim = to_merge.SubgroupManualDim(); in MergeShardingIfCompatible()
140 int64_t dst_man_dim = dst->SubgroupManualDim(); in MergeShardingIfCompatible()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2914 if (i == hlo->sharding().SubgroupManualDim()) { in HandleAllReduce()