Home
last modified time | relevance | path

Searched refs:position_and_chunk (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc2940 for (const auto& position_and_chunk : alternate_memory_assignments_) { in ExportAndColorBuffers() local
2941 const HloPosition& defining_position = position_and_chunk.first; in ExportAndColorBuffers()
2942 const Chunk& chunk = position_and_chunk.second; in ExportAndColorBuffers()
2995 for (auto& position_and_chunk : alternate_memory_assignments_) { in RemoveAssignmentForInstruction() local
2996 const HloPosition& position = position_and_chunk.first; in RemoveAssignmentForInstruction()
3000 position_and_chunk = alternate_memory_assignments_.back(); in RemoveAssignmentForInstruction()
3319 for (const auto& position_and_chunk : preset_assignments_->chunks()) { in VerifyAndExportHeapSimulatorTrace() local
3320 const HloPosition& position = position_and_chunk.first; in VerifyAndExportHeapSimulatorTrace()
3321 const Chunk& chunk = position_and_chunk.second; in VerifyAndExportHeapSimulatorTrace()
Dbuffer_assignment.cc1382 for (auto& position_and_chunk : preset_assignments_->chunks()) { in AssignPresetBuffers() local
1383 const HloPosition& defining_position = position_and_chunk.first; in AssignPresetBuffers()
1388 const HeapSimulator::Chunk& chunk = position_and_chunk.second; in AssignPresetBuffers()
Dmemory_space_assignment_test.cc163 for (auto& position_and_chunk : preset_assignments->chunks()) { in CheckPresetAssignments() local
164 HloPosition position = position_and_chunk.first; in CheckPresetAssignments()
3937 for (const auto& position_and_chunk : preset_assignments->chunks()) { in TEST_P() local
3938 const HloPosition& position = position_and_chunk.first; in TEST_P()