Searched refs:EvenlyPartitions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 85 bool EvenlyPartitions(const Shape& shape, const HloSharding& sharding);
|
| D | spmd_partitioner_util.cc | 127 bool EvenlyPartitions(const Shape& shape, const HloSharding& sharding) { in EvenlyPartitions() function
130 if (!EvenlyPartitions(ShapeUtil::GetTupleElementShape(shape, i), in EvenlyPartitions()
281 if (EvenlyPartitions(base_shape, sharding)) { in GetPaddedShapeForUnevenPartitioning()
|
| D | spmd_partitioner.cc | 485 if (sharding.IsReplicated() || EvenlyPartitions(base_shape_, sharding)) { in PadWithValue()
1779 if (!EvenlyPartitions(hlo->shape(), input_partitioned.sharding())) { in HandleCustomCall()
2268 (!EvenlyPartitions(hlo->shape(), hlo->sharding()) || in HandleConstant()
2508 if (EvenlyPartitions(shape, sharding)) { in HandleInfeed()
2911 if (EvenlyPartitions(shape, sharding)) { in HandleOutfeed()
|