HomeSort by relevance Sort by last modified time
    Searched refs:thunk_launch_order_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_hlo_schedule.h 50 return thunk_launch_order_;
63 std::vector<HloInstruction*> thunk_launch_order_; member in class:xla::gpu::GpuHloSchedule
gpu_hlo_schedule.cc 197 // Initialize thunk_launch_order_, the total order of thunk launches.
208 schedule->thunk_launch_order_ = sequence.instructions();
211 BFSLaunchOrder(entry_computation, &schedule->thunk_launch_order_);
215 &module, stream_assignment, schedule->thunk_launch_order_);

Completed in 445 milliseconds