Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dmemory_space_assignment.cc3835 alternate_memory_assignments_.emplace_back( in Process()
3858 for (const auto& position_and_chunk : alternate_memory_assignments_) { in ExportAndColorBuffers()
3923 auto it = alternate_memory_assignments_.begin(); in RemoveAssignmentForInstruction()
3924 auto end = alternate_memory_assignments_.end(); in RemoveAssignmentForInstruction()
3930 alternate_memory_assignments_.pop_back(); in RemoveAssignmentForInstruction()
3934 *it = alternate_memory_assignments_.back(); in RemoveAssignmentForInstruction()
3935 alternate_memory_assignments_.pop_back(); in RemoveAssignmentForInstruction()
3936 end = alternate_memory_assignments_.end(); in RemoveAssignmentForInstruction()
H A Dmemory_space_assignment.h997 std::vector<std::pair<HloPosition, Chunk>> alternate_memory_assignments_; variable