Home
last modified time | relevance | path

Searched refs:preset_assignments_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h629 preset_assignments_(std::move(preset_assignments)) {} in BufferAssigner()
708 std::unique_ptr<PresetAssignments> preset_assignments_; variable
Dmemory_space_assignment.cc2686 return std::move(preset_assignments_); in RunMemorySpaceAssignment()
2954 preset_assignments_->add_chunk(defining_position, chunk); in ExportAndColorBuffers()
2959 if (!preset_assignments_->chunks().empty()) { in ExportAndColorBuffers()
2960 preset_assignments_ in ExportAndColorBuffers()
2966 for (auto& pair : preset_assignments_->assignment_informations()) { in ExportAndColorBuffers()
2973 preset_assignments_->chunks()) { in ExportAndColorBuffers()
3319 for (const auto& position_and_chunk : preset_assignments_->chunks()) { in VerifyAndExportHeapSimulatorTrace()
3430 &preset_assignments_ in VerifyAndExportHeapSimulatorTrace()
Dbuffer_assignment.cc1359 if (!preset_assignments_) { in AssignPresetBuffers()
1366 for (auto& color_and_info : preset_assignments_->assignment_informations()) { in AssignPresetBuffers()
1382 for (auto& position_and_chunk : preset_assignments_->chunks()) { in AssignPresetBuffers()
1402 preset_assignments_ = {}; in AssignPresetBuffers()
Dmemory_space_assignment.h834 preset_assignments_(absl::make_unique<PresetAssignments>()) { in MemorySpaceAssignment()
882 std::unique_ptr<PresetAssignments> preset_assignments_; variable