Searched defs:preset_assignments (Results 1 – 4 of 4) sorted by relevance
169 std::unique_ptr<PresetAssignments> preset_assignments = in AssignMemorySpace() local181 void CheckPresetAssignments(const PresetAssignments* preset_assignments) { in CheckPresetAssignments()267 int64_t GetAlternateMemoryOffset(const PresetAssignments& preset_assignments, in GetAlternateMemoryOffset()478 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local2413 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local3892 std::unique_ptr<PresetAssignments> preset_assignments = AssignMemorySpace( in TEST_P() local5694 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local6498 std::unique_ptr<PresetAssignments> preset_assignments = AssignMemorySpace( in TEST_P() local6549 std::unique_ptr<PresetAssignments> preset_assignments = AssignMemorySpace( in TEST_P() local6570 auto preset_assignments = AssignMemorySpace( in TEST_P() local[all …]
633 std::unique_ptr<memory_space_assignment::PresetAssignments> in BufferAssigner()
165 HloModule* module, std::unique_ptr<PresetAssignments> preset_assignments, in RunBufferAssignmentWithPresetAssignments()748 auto preset_assignments = std::make_unique<PresetAssignments>(); in TEST_F() local855 auto preset_assignments = std::make_unique<PresetAssignments>(); in TEST_F() local
963 std::unique_ptr<PresetAssignments> preset_assignments) { in Run()