Searched defs:slice_offsets (Results 1 – 5 of 5) sorted by relevance
98 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), zero_s32); in PadEachPartitionWithHaloExchange() local
407 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in TileToPartialReplicateHaloExchange() local516 std::vector<HloInstruction*> slice_offsets(concat->shape().rank(), in PadFromPartialReplicateShape() local966 std::vector<HloInstruction*> slice_offsets(base_shape.rank(), zero); in ExchangeHaloAndGetValidData() local
4180 std::vector<HloInstruction*> slice_offsets; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() member4233 HloInstruction* dus) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()4338 HloInstruction* last_iter_result) -> StatusOr<HloInstruction*> { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4148 std::vector<HloInstruction*> slice_offsets(shard_shape.rank(), zero); in HandleSelectAndScatter() local
57298 pub fn slice_offsets(mut self, slice_offsets: &'a [u32]) -> Self { in slice_offsets() method