Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertKernelFuncToBlob.cpp47 class GpuKernelToBlobPass class
48 : public PassWrapper<GpuKernelToBlobPass, OperationPass<gpu::GPUModuleOp>> {
50 GpuKernelToBlobPass(LoweringCallback loweringCallback, in GpuKernelToBlobPass() function in __anon2141a3090111::GpuKernelToBlobPass
104 GpuKernelToBlobPass::translateModuleToISA(llvm::Module &module, in translateModuleToISA()
119 OwnedBlob GpuKernelToBlobPass::convertModuleToBlob(llvm::Module &llvmModule, in convertModuleToBlob()
146 StringAttr GpuKernelToBlobPass::translateGPUModuleToBinaryAnnotation( in translateGPUModuleToBinaryAnnotation()
160 return std::make_unique<GpuKernelToBlobPass>(loweringCallback, blobGenerator, in createConvertGPUKernelToBlobPass()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dgpu_kernel_to_blob_pass.cc51 class GpuKernelToBlobPass class
52 : public GpuKernelToBlobPassBase<GpuKernelToBlobPass> {
54 GpuKernelToBlobPass(mlir::StringRef blob_annotation, in GpuKernelToBlobPass() function in mlir::kernel_gen::transforms::__anond5a855260111::GpuKernelToBlobPass
257 return std::make_unique<GpuKernelToBlobPass>( in CreateGpuKernelToBlobPass()
Dpasses.td74 def GpuKernelToBlobPass : Pass<"gpu-kernel-to-blob", "gpu::GPUModuleOp"> {