Searched refs:MakeTiledPartitionOrdinals (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 111 std::vector<HloInstruction*> MakeTiledPartitionOrdinals(
|
| D | fft_handler.cc | 102 MakeTiledPartitionOrdinals(sharding, partition_id, b); in PadEachPartitionWithHaloExchange()
|
| D | convolution_handler.cc | 349 MakeTiledPartitionOrdinals(lhs.sharding(), partition_id, b); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
656 MakeTiledPartitionOrdinals(lhs.sharding(), partition_id, b); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
|
| D | spmd_partitioner.cc | 567 MakeTiledPartitionOrdinals(target, state_.partition_id, state_.b); in ReshardAsWindowedInput()
1544 .Calculate(MakeTiledPartitionOrdinals(sharding, partition_id_, in HandleConcatenate()
1567 auto start_indices = MakeTiledPartitionOrdinals( in HandleConcatenate()
2156 MakeTiledPartitionOrdinals(sharding, partition_id_, &b_); in HandleIota()
2377 MakeTiledPartitionOrdinals(hlo->sharding(), partition_id_, &b_); in HandleDynamicUpdateSlice()
3208 MakeTiledPartitionOrdinals(hlo->sharding(), partition_id_, &b_); in HandleSelectAndScatter()
|
| D | spmd_partitioner_util.cc | 239 std::vector<HloInstruction*> MakeTiledPartitionOrdinals( in MakeTiledPartitionOrdinals() function
425 MakeTiledPartitionOrdinals(dst_sharding, partition_id, b); in TileToPartialReplicateHaloExchange()
517 MakeTiledPartitionOrdinals(src_sharding, partition_id, b); in PadFromPartialReplicateShape()
|