Searched defs:preset_assignments (Results 1 – 4 of 4) sorted by relevance
625 std::unique_ptr<PresetAssignments> preset_assignments) in BufferAssigner()
146 std::unique_ptr<PresetAssignments> preset_assignments = in AssignMemorySpace() local157 void CheckPresetAssignments(const PresetAssignments* preset_assignments) { in CheckPresetAssignments()235 int64 GetAlternateMemoryOffset(const PresetAssignments& preset_assignments, in GetAlternateMemoryOffset()445 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local3924 std::unique_ptr<PresetAssignments> preset_assignments = in TEST_P() local4765 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local
148 HloModule* module, std::unique_ptr<PresetAssignments> preset_assignments, in RunBufferAssignmentWithPresetAssignments()729 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local836 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local
916 std::unique_ptr<PresetAssignments> preset_assignments) { in Run()