Searched refs:operand_slice (Results 1 – 3 of 3) sorted by relevance
96 const HloInstruction* operand_slice = candidate->operand(i); in IsSimilarOperationOnSlices() local97 if (operand_slice->opcode() != HloOpcode::kSlice || in IsSimilarOperationOnSlices()98 operand_slice->operand(0) != in IsSimilarOperationOnSlices()100 !SameSliceConfiguration(operand_slice0, operand_slice)) { in IsSimilarOperationOnSlices()182 const HloInstruction* operand_slice = user->operand(0); in SinkSlices() local184 computation->AddInstruction(operand_slice->CloneWithNewOperands( in SinkSlices()
1169 for (const auto& operand_slice : in AssignSingleHloBuffer() local1172 assignment->GetMutableAllocation(operand_slice.index()); in AssignSingleHloBuffer()
4202 HloInstruction* operand_slice = slice->mutable_operand(0); in HandleSlice() local4206 new_slice_starts[i] += operand_slice->slice_starts(i); in HandleSlice()4207 new_slice_limits[i] += operand_slice->slice_starts(i); in HandleSlice()4211 slice->shape(), operand_slice->mutable_operand(0), in HandleSlice()