Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc720 copy_start_schedule_after() == other.copy_start_schedule_after() && in operator ==()
3147 first->copy_start_schedule_after()) < in ScheduleAsynchronousCopies()
3149 second->copy_start_schedule_after()); in ScheduleAsynchronousCopies()
3156 int64 copy_start_schedule_after = in ScheduleAsynchronousCopies() local
3157 copy_allocation->copy_start_schedule_after(); in ScheduleAsynchronousCopies()
3161 flattened_instructions_[copy_start_schedule_after]->parent()) { in ScheduleAsynchronousCopies()
3162 VLOG(4) << "Delaying CopyStart (" << copy_start_schedule_after << " to " in ScheduleAsynchronousCopies()
3163 << (copy_start_schedule_after + 1) << ") for " in ScheduleAsynchronousCopies()
3167 ++copy_start_schedule_after); in ScheduleAsynchronousCopies()
3170 schedule_after_[copy_allocation->copy_start_schedule_after()].push_back( in ScheduleAsynchronousCopies()
Dmemory_space_assignment.h627 int64 copy_start_schedule_after() const { in copy_start_schedule_after() function
634 void set_copy_start_schedule_after(int64 copy_start_schedule_after) { in set_copy_start_schedule_after() argument
635 copy_start_schedule_after_ = copy_start_schedule_after; in set_copy_start_schedule_after()