Home
last modified time | relevance | path

Searched refs:KernelThunk (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dkernel_thunk.h45 class KernelThunk : public Thunk {
50 KernelThunk(ThunkInfo thunk_info,
53 KernelThunk(const KernelThunk&) = delete;
54 KernelThunk& operator=(const KernelThunk&) = delete;
55 ~KernelThunk() override = default;
Dkernel_thunk.cc36 KernelThunk::KernelThunk(ThunkInfo thunk_info, in KernelThunk() function in xla::gpu::KernelThunk
43 Status KernelThunk::Initialize(const GpuExecutable& executable, in Initialize()
65 void KernelThunk::SetLaunchDimensions(const LaunchDimensions& launch_dims) { in SetLaunchDimensions()
87 Status KernelThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream()
Dir_emitter_unnested.h235 KernelThunk* thunk, int unroll_factor, bool few_waves = false);
641 std::unique_ptr<KernelThunk> BuildKernelThunkFromBufferSlices(
652 std::unique_ptr<KernelThunk> BuildKernelThunk(
655 std::unique_ptr<KernelThunk> BuildKernelThunkForMlirImpl(
660 StatusOr<std::unique_ptr<KernelThunk>> BuildKernelThunkForMlir(
665 StatusOr<std::unique_ptr<KernelThunk>> BuildKernelThunkForMlir(
Dir_emitter_unnested.cc161 KernelThunk* kernel_thunk = static_cast<KernelThunk*>(thunk); in UpdateLaunchDimensions()
1772 TF_ASSIGN_OR_RETURN(std::unique_ptr<KernelThunk> kernel_thunk_ptr, in EmitLoopFusionFromMlir()
3340 std::unique_ptr<KernelThunk>
3451 return absl::make_unique<KernelThunk>(thunk_info, non_constant_buffers, in BuildKernelThunkFromBufferSlices()
3455 std::unique_ptr<KernelThunk> IrEmitterUnnested::BuildKernelThunk( in BuildKernelThunk()
3483 std::unique_ptr<KernelThunk> IrEmitterUnnested::BuildKernelThunkForMlirImpl( in BuildKernelThunkForMlirImpl()
3513 StatusOr<std::unique_ptr<KernelThunk>>
3535 StatusOr<std::unique_ptr<KernelThunk>>
3666 std::unique_ptr<KernelThunk> kernel_thunk, in BuildInitializerThunkForMlir()
3711 std::unique_ptr<KernelThunk> kernel_thunk, in BuildFusedInitializerThunkForMlir()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/g3doc/
Dxla_gpu_codegen.md225 * KernelThunk, which launches a kernel.