Searched defs:thunk_schedule (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc116 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 Dgpu_compiler.cc56 #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