Searched defs:partition_ordinals (Results 1 – 4 of 4) sorted by relevance
101 auto partition_ordinals = in PadEachPartitionWithHaloExchange() local
348 auto partition_ordinals = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local655 auto partition_ordinals = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
566 auto partition_ordinals = in ReshardAsWindowedInput() local2155 auto partition_ordinals = in HandleIota() local2376 auto partition_ordinals = in HandleDynamicUpdateSlice() local3207 auto partition_ordinals = in HandleSelectAndScatter() local
424 auto partition_ordinals = in TileToPartialReplicateHaloExchange() local516 auto partition_ordinals = in PadFromPartialReplicateShape() local