Searched defs:preset_assignments (Results 1 – 4 of 4) sorted by relevance
629 std::unique_ptr<memory_space_assignment::PresetAssignments> in BufferAssigner()
164 std::unique_ptr<PresetAssignments> preset_assignments = in AssignMemorySpace() local175 void CheckPresetAssignments(const PresetAssignments* preset_assignments) { in CheckPresetAssignments()253 int64 GetAlternateMemoryOffset(const PresetAssignments& preset_assignments, in GetAlternateMemoryOffset()464 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local3943 std::unique_ptr<PresetAssignments> preset_assignments = in TEST_P() local5510 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local5919 auto preset_assignments = AssignMemorySpace( in TEST_P() local
150 HloModule* module, std::unique_ptr<PresetAssignments> preset_assignments, in RunBufferAssignmentWithPresetAssignments()731 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local838 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local
922 std::unique_ptr<PresetAssignments> preset_assignments) { in Run()