Home
last modified time | relevance | path

Searched refs:RequiredMemoryAssignment (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.h1054 struct RequiredMemoryAssignment { struct
1059 bool equals_ignoring_time(const RequiredMemoryAssignment& other) const { in equals_ignoring_time() argument
1063 bool operator==(const RequiredMemoryAssignment& other) const {
1068 bool operator!=(const RequiredMemoryAssignment& other) const {
1200 absl::optional<RequiredMemoryAssignment> RequiredMemoryAssignmentAt(
1205 absl::optional<RequiredMemoryAssignment> AliasedRequiredAssignmentForUse(
1338 std::vector<std::pair<const HloValue*, RequiredMemoryAssignment>>
1347 absl::flat_hash_map<const HloValue*, std::vector<RequiredMemoryAssignment>>
Dmemory_space_assignment.cc1633 absl::optional<AlternateMemoryBestFitHeap::RequiredMemoryAssignment>
1637 absl::optional<RequiredMemoryAssignment> required_assignment_at_time; in RequiredMemoryAssignmentAt()
1639 for (const RequiredMemoryAssignment& required_assignment : in RequiredMemoryAssignmentAt()
1651 absl::optional<AlternateMemoryBestFitHeap::RequiredMemoryAssignment>
1654 absl::optional<RequiredMemoryAssignment> required_assignment; in AliasedRequiredAssignmentForUse()
1661 absl::optional<RequiredMemoryAssignment> required_assignment_for_alias = in AliasedRequiredAssignmentForUse()
1704 RequiredMemoryAssignment required_assignment{memory_space, time, offset}; in AddRequiredAssignment()
1860 const RequiredMemoryAssignment& required_assignment = in UncommitPendingChunks()