Searched refs:WhileThunk (Results 1 – 3 of 3) sorted by relevance
39 class WhileThunk : public Thunk {42 WhileThunk(ThunkInfo thunk_info,48 WhileThunk(const WhileThunk&) = delete;49 WhileThunk& operator=(const WhileThunk&) = delete;
26 WhileThunk::WhileThunk( in WhileThunk() function in xla::gpu::WhileThunk46 Status WhileThunk::Initialize(const GpuExecutable& executable, in Initialize()54 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
3861 return std::unique_ptr<Thunk>(new WhileThunk( in BuildWhileThunk()