Searched refs:GetUniqueTopLevelSlice (Results 1 – 5 of 5) sorted by relevance
253 *buffers.GetUniqueTopLevelSlice(hlo).ConsumeValueOrDie().allocation(); in GetAssignedInputAllocation()275 return *buffers.GetUniqueTopLevelSlice(hlo) in GetTopLevelAllocation()325 assignment.GetUniqueTopLevelSlice(instruction).ConsumeValueOrDie()); in BuffersDistinct()331 if (a_slices.contains(assignment.GetUniqueTopLevelSlice(instruction) in BuffersDistinct()1788 assignment->GetUniqueTopLevelSlice(dot_ab).ConsumeValueOrDie(); in TEST_F()1790 assignment->GetUniqueTopLevelSlice(dot_bc).ConsumeValueOrDie(); in TEST_F()1801 slice_ab = assignment->GetUniqueTopLevelSlice(dot_ab).ConsumeValueOrDie(); in TEST_F()1802 slice_bc = assignment->GetUniqueTopLevelSlice(dot_bc).ConsumeValueOrDie(); in TEST_F()2685 auto* root_alloc = assignment->GetUniqueTopLevelSlice(while1_out) in TEST_F()2738 assignment->GetUniqueTopLevelSlice(dus9).ConsumeValueOrDie(); in TEST_F()[all …]
421 StatusOr<BufferAllocation::Slice> GetUniqueTopLevelSlice(
504 StatusOr<BufferAllocation::Slice> BufferAssignment::GetUniqueTopLevelSlice( in GetUniqueTopLevelSlice() function in xla::BufferAssignment552 return GetUniqueTopLevelSlice( in GetUniqueTopLevelOutputSlice()
170 assignment_.GetUniqueTopLevelSlice(computation->root_instruction())); in EmitComputation()175 assignment_.GetUniqueTopLevelSlice(param)); in EmitComputation()3116 assignment_.GetUniqueTopLevelSlice(op)); in EmitTargetAddressForOp()3318 assignment_.GetUniqueTopLevelSlice(root_inst).ValueOrDie(); in GetBufferForGlobalCallReturnValue()
927 assignment_.GetUniqueTopLevelSlice(instr)); in EmitConstant()