Home
last modified time | relevance | path

Searched refs:CreateGpuKernelToBlobPass (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dpasses.h70 std::unique_ptr<OperationPass<gpu::GPUModuleOp>> CreateGpuKernelToBlobPass(
H A Dgpu_kernel_to_blob_pass.cc259 std::unique_ptr<OperationPass<gpu::GPUModuleOp>> CreateGpuKernelToBlobPass( in CreateGpuKernelToBlobPass() function
H A Dpasses.td94 let constructor = "transforms::CreateGpuKernelToBlobPass()";
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc381 kernel_pm.addPass(mlir::kernel_gen::transforms::CreateGpuKernelToBlobPass( in GenerateDeviceCode()