HomeSort by relevance Sort by last modified time
    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