Searched refs:AllocationValue (Results 1 – 2 of 2) sorted by relevance
726 class AllocationValue {750 AllocationValue(const HloValue* value, const HloPosition& position, in AllocationValue() function945 using AllocationValue = MemorySpaceAssignment::AllocationValue; variable982 std::vector<AllocationValue>& allocation_values) const;989 std::vector<MemorySpaceAssignment::AllocationValue>& allocation_values);993 void FindAliases(std::vector<AllocationValue>* allocation_values,1046 const MemorySpaceAssignment::AllocationValue::Use* use;1047 MemorySpaceAssignment::AllocationValue* allocation_value;1147 bool IsUseAllowedInAlternateMemory(const AllocationValue& value,1155 absl::Span<AllocationValue> allocation_values);[all …]
724 std::string MemorySpaceAssignment::AllocationValue::ToString() const { in ToString()735 std::string MemorySpaceAssignment::AllocationValue::ToShortString() const { in ToShortString()743 std::vector<AllocationValue>& allocation_values) const { in CreateAllocationValues()795 AllocationValue* last_allocation_value = nullptr; in CreateAllocationValues()797 AllocationValue* allocation_value = &allocation_values.at(i); in CreateAllocationValues()815 std::vector<AllocationValue>* allocation_values, in FindAliases()817 absl::flat_hash_map<const HloInstruction*, const AllocationValue*> in FindAliases()819 for (AllocationValue& value : *allocation_values) { in FindAliases()828 AllocationValue::Use* use) { in FindAliases()837 for (AllocationValue& value : *allocation_values) { in FindAliases()[all …]