Searched refs:GetUniqueTopLevelSlice (Results 1 – 6 of 6) sorted by relevance
246 *buffers.GetUniqueTopLevelSlice(hlo).ConsumeValueOrDie().allocation(); in GetAssignedInputAllocation()268 return *buffers.GetUniqueTopLevelSlice(hlo) in GetTopLevelAllocation()318 assignment.GetUniqueTopLevelSlice(instruction).ConsumeValueOrDie()); in BuffersDistinct()324 if (a_slices.contains(assignment.GetUniqueTopLevelSlice(instruction) in BuffersDistinct()1616 assignment->GetUniqueTopLevelSlice(dot_ab).ConsumeValueOrDie(); in TEST_F()1618 assignment->GetUniqueTopLevelSlice(dot_bc).ConsumeValueOrDie(); in TEST_F()1629 slice_ab = assignment->GetUniqueTopLevelSlice(dot_ab).ConsumeValueOrDie(); in TEST_F()1630 slice_bc = assignment->GetUniqueTopLevelSlice(dot_bc).ConsumeValueOrDie(); in TEST_F()2504 auto* root_alloc = assignment->GetUniqueTopLevelSlice(while1_out) in TEST_F()2558 assignment->GetUniqueTopLevelSlice(dus9).ConsumeValueOrDie(); in TEST_F()[all …]
401 StatusOr<BufferAllocation::Slice> GetUniqueTopLevelSlice(
442 StatusOr<BufferAllocation::Slice> BufferAssignment::GetUniqueTopLevelSlice( in GetUniqueTopLevelSlice() function in xla::BufferAssignment490 return GetUniqueTopLevelSlice( in GetUniqueTopLevelOutputSlice()
77 ->GetUniqueTopLevelSlice(non_io_hlo->LatestNonGteAncestor()) in EmitBasePointersForHlos()
695 buffer_assignment.GetUniqueTopLevelSlice(copy->operand(0)).ok()) { in HandleCopy()746 .GetUniqueTopLevelSlice(tuple_element) in HandleTuple()
125 assignment_.GetUniqueTopLevelSlice(computation->root_instruction())); in EmitComputation()130 assignment_.GetUniqueTopLevelSlice(param)); in EmitComputation()2935 assignment_.GetUniqueTopLevelSlice(op)); in EmitTargetAddressForOp()3101 assignment_.GetUniqueTopLevelSlice(root_inst).ValueOrDie(); in GetBufferForGlobalCallReturnValue()