OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:KernelThunk
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
kernel_thunk.cc
30
KernelThunk
::
KernelThunk
(absl::Span<const BufferAllocation* const> args,
39
Status
KernelThunk
::Initialize(const GpuExecutable& executable,
69
void
KernelThunk
::SetLaunchDimensions(const LaunchDimensions& launch_dims) {
74
Status
KernelThunk
::ExecuteOnStream(const BufferAllocations& buffer_allocations,
kernel_thunk.h
45
class
KernelThunk
: public Thunk {
50
KernelThunk
(absl::Span<const BufferAllocation* const> args,
53
KernelThunk
(const
KernelThunk
&) = delete;
54
KernelThunk
& operator=(const
KernelThunk
&) = delete;
55
~
KernelThunk
() override = default;
Completed in 937 milliseconds