Searched refs:preset_assignments (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | buffer_assignment.h | 620 std::unique_ptr<PresetAssignments> preset_assignments = {});
625 std::unique_ptr<PresetAssignments> preset_assignments) in BufferAssigner() argument
629 preset_assignments_(std::move(preset_assignments)) {} in BufferAssigner()
|
| D | buffer_assignment_test.cc | 148 HloModule* module, std::unique_ptr<PresetAssignments> preset_assignments, in RunBufferAssignmentWithPresetAssignments() argument
157 /*can_share_buffer=*/nullptr, std::move(preset_assignments)) in RunBufferAssignmentWithPresetAssignments()
729 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local
730 preset_assignments->add_chunk({mul, {}}, {/*offset=*/100, /*size=*/400}); in TEST_F()
731 preset_assignments->add_chunk({add, {}}, {/*offset=*/550, /*size=*/400}); in TEST_F()
732 preset_assignments->assignment_information_for_space(/*memory_space=*/1) in TEST_F()
736 module.get(), std::move(preset_assignments)); in TEST_F()
836 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F() local
837 preset_assignments->add_chunk({negate, {}}, {/*offset=*/100, /*size=*/40}); in TEST_F()
838 preset_assignments->assignment_information_for_space(/*memory_space=*/1) in TEST_F()
[all …]
|
| D | memory_space_assignment_test.cc | 146 std::unique_ptr<PresetAssignments> preset_assignments = in AssignMemorySpace() local
153 CheckPresetAssignments(preset_assignments.get()); in AssignMemorySpace()
154 return preset_assignments; in AssignMemorySpace()
157 void CheckPresetAssignments(const PresetAssignments* preset_assignments) { in CheckPresetAssignments() argument
163 for (auto& position_and_chunk : preset_assignments->chunks()) { in CheckPresetAssignments()
235 int64 GetAlternateMemoryOffset(const PresetAssignments& preset_assignments, in GetAlternateMemoryOffset() argument
243 for (auto& pos_and_chunk : preset_assignments.chunks()) {
445 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P() local
460 EXPECT_EQ(preset_assignments->chunks().size(), 3); in TEST_P()
461 EXPECT_EQ(preset_assignments->assignment_informations().size(), 1); in TEST_P()
[all …]
|
| D | buffer_assignment.cc | 916 std::unique_ptr<PresetAssignments> preset_assignments) { in Run() argument
918 must_not_live_out, std::move(preset_assignments)); in Run()
|