Searched refs:operand_slice (Results 1 – 2 of 2) sorted by relevance
1052 for (const auto& operand_slice : in AssignBuffersForComputation() local1055 assignment->GetMutableAllocation(operand_slice.index()); in AssignBuffersForComputation()
2828 HloInstruction* operand_slice = slice->mutable_operand(0); in HandleSlice() local2832 new_slice_starts[i] += operand_slice->slice_starts(i); in HandleSlice()2833 new_slice_limits[i] += operand_slice->slice_starts(i); in HandleSlice()2837 slice->shape(), operand_slice->mutable_operand(0), in HandleSlice()