Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc3137 auto copy_allocation = static_cast<CopyAllocation*>(allocation.get()); in ScheduleAsynchronousCopies() local
3138 if (copy_allocation->memory_space() == memory_space) { in ScheduleAsynchronousCopies()
3139 copy_allocations.push_back(copy_allocation); in ScheduleAsynchronousCopies()
3153 for (CopyAllocation* copy_allocation : copy_allocations) { in ScheduleAsynchronousCopies()
3157 copy_allocation->copy_start_schedule_after(); in ScheduleAsynchronousCopies()
3160 while (copy_allocation->defining_position().instruction->parent() != in ScheduleAsynchronousCopies()
3164 << copy_allocation->copy_start()->ToString() in ScheduleAsynchronousCopies()
3166 copy_allocation->set_copy_start_schedule_after( in ScheduleAsynchronousCopies()
3170 schedule_after_[copy_allocation->copy_start_schedule_after()].push_back( in ScheduleAsynchronousCopies()
3171 copy_allocation->copy_start()); in ScheduleAsynchronousCopies()
[all …]