Home
last modified time | relevance | path

Searched refs:operand_slice (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dslice_sinker.cc96 const HloInstruction* operand_slice = candidate->operand(i); in IsSimilarOperationOnSlices() local
97 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() local
184 computation->AddInstruction(operand_slice->CloneWithNewOperands( in SinkSlices()
Dbuffer_assignment.cc1169 for (const auto& operand_slice : in AssignSingleHloBuffer() local
1172 assignment->GetMutableAllocation(operand_slice.index()); in AssignSingleHloBuffer()
Dalgebraic_simplifier.cc4202 HloInstruction* operand_slice = slice->mutable_operand(0); in HandleSlice() local
4206 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()