Searched defs:thunk_schedule (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_executable.cc | 116 std::unique_ptr<const ThunkSchedule> thunk_schedule, 126 thunk_schedule_(std::move(thunk_schedule)), 113 GpuExecutable( const string& ptx, const std::vector<uint8>& cubin, std::pair<int, int> compute_capability, std::unique_ptr<const ThunkSchedule> thunk_schedule, std::unique_ptr<const HloModule> hlo_module, std::unique_ptr<const BufferAssignment> assignment, std::unique_ptr<HloProfilePrinterData> hlo_profile_printer_data, std::unique_ptr<HloProfileIndexMap> hlo_profile_index_map) argument
|
H A D | gpu_compiler.cc | 56 #include "tensorflow/compiler/xla/service/gpu/thunk_schedule.h" 650 auto thunk_schedule = MakeUnique<ThunkSchedule>( local 654 XLA_VLOG_LINES(2, thunk_schedule->ToString()); 668 ptx, cubin, {cc_major, cc_minor}, std::move(thunk_schedule),
|
Completed in 104 milliseconds