Home
last modified time | relevance | path

Searched refs:ForThunk (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfor_thunk.h32 class ForThunk : public Thunk {
34 ForThunk(ThunkInfo thunk_info, const int64 loop_limit,
37 ForThunk(const ForThunk&) = delete;
38 ForThunk& operator=(const ForThunk&) = delete;
Dfor_thunk.cc26 ForThunk::ForThunk(ThunkInfo thunk_info, const int64 loop_limit, in ForThunk() function in xla::gpu::ForThunk
38 Status ForThunk::Initialize(const GpuExecutable& executable, in Initialize()
44 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dir_emitter_unnested.cc3888 return std::unique_ptr<Thunk>(new ForThunk( in BuildForThunk()