Home
last modified time | relevance | path

Searched refs:PresetAssignments (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.h31 class PresetAssignments {
40 PresetAssignments() = default;
794 static StatusOr<std::unique_ptr<PresetAssignments>> Run(
811 virtual StatusOr<std::unique_ptr<PresetAssignments>> RunMemorySpaceAssignment(
834 preset_assignments_(absl::make_unique<PresetAssignments>()) { in MemorySpaceAssignment()
882 std::unique_ptr<PresetAssignments> preset_assignments_;
Dbuffer_assignment.h620 std::unique_ptr<PresetAssignments> preset_assignments = {});
625 std::unique_ptr<PresetAssignments> preset_assignments) in BufferAssigner()
708 std::unique_ptr<PresetAssignments> preset_assignments_;
Dbuffer_assignment_test.cc148 HloModule* module, std::unique_ptr<PresetAssignments> preset_assignments, in RunBufferAssignmentWithPresetAssignments()
699 TEST_F(BufferAssignmentTest, PresetAssignments) { in TEST_F() argument
729 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F()
836 auto preset_assignments = absl::make_unique<PresetAssignments>(); in TEST_F()
Dmemory_space_assignment_test.cc46 std::unique_ptr<PresetAssignments> AssignMemorySpaceUsingCostAnalysis( in AssignMemorySpaceUsingCostAnalysis()
72 std::unique_ptr<PresetAssignments> AssignMemorySpace( in AssignMemorySpace()
83 std::unique_ptr<PresetAssignments> AssignMemorySpace( in AssignMemorySpace()
146 std::unique_ptr<PresetAssignments> preset_assignments = in AssignMemorySpace()
157 void CheckPresetAssignments(const PresetAssignments* preset_assignments) { in CheckPresetAssignments()
235 int64 GetAlternateMemoryOffset(const PresetAssignments& preset_assignments, in GetAlternateMemoryOffset()
3924 std::unique_ptr<PresetAssignments> preset_assignments = in TEST_P()
Dmemory_space_assignment.cc2646 /*static*/ StatusOr<std::unique_ptr<PresetAssignments>>
2662 StatusOr<std::unique_ptr<PresetAssignments>>
Dbuffer_assignment.cc916 std::unique_ptr<PresetAssignments> preset_assignments) { in Run()