Searched refs:zero_s32 (Results 1 – 4 of 4) sorted by relevance
98 auto zero_s32 = in PadEachPartitionWithHaloExchange() local100 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), zero_s32); in PadEachPartitionWithHaloExchange()
481 auto zero_s32 = b->AddInstruction( in TileToPartialReplicateHaloExchange() local493 zero_s32); in TileToPartialReplicateHaloExchange()593 auto zero_s32 = b->AddInstruction( in PadFromPartialReplicateShape() local601 zero_s32); in PadFromPartialReplicateShape()
352 auto zero_s32 = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local355 output_base_shape.rank(), zero_s32); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
749 HloInstruction* zero_s32 = comp->AddInstruction( in RewriteInputWithDynamicPadding() local758 std::vector<HloInstruction*> start_indices(input->shape().rank(), zero_s32); in RewriteInputWithDynamicPadding()