Searched refs:HasAllocation (Results 1 – 3 of 3) sorted by relevance
394 bool BufferAssignment::HasAllocation(const HloValue& value) const { in HasAllocation() function in xla::BufferAssignment398 bool BufferAssignment::HasAllocation(const HloBuffer& buffer) const { in HasAllocation() function in xla::BufferAssignment404 CHECK(HasAllocation(value)); in GetAssignedAllocation()423 if (HasAllocation(*value)) { in GetAllSlices()442 if (!HasAllocation(*value)) { in GetInstructionAllocation()480 if (HasAllocation(*value)) { in GetUniqueSlice()878 if (HasAllocation(*value)) { in ToProto()987 CHECK(!assignment->HasAllocation(hlo_buffer)) in MaybeAssignBuffer()1196 if (!assignment->HasAllocation(*hlo_buffer) && in AssignSingleHloBuffer()1225 if (!assignment->HasAllocation(*hlo_buffer)) { in AssignSingleHloBuffer()[all …]
380 bool HasAllocation(const HloValue& value) const;382 bool HasAllocation(const HloBuffer& buffer) const;
154 if (assignment.HasAllocation(*buffer)) { in GetNoaliasMetadataForBuffer()