Home
last modified time | relevance | path

Searched defs:kernelModule (Results 1 – 5 of 5) sorted by relevance

/external/llvm-project/mlir/tools/mlir-spirv-cpu-runner/
Dmlir-spirv-cpu-runner.cpp58 std::unique_ptr<llvm::Module> kernelModule = in convertMLIRModule() local
/external/llvm-project/mlir/lib/Dialect/GPU/Transforms/
DKernelOutlining.cpp257 auto kernelModule = createKernelModule(outlinedFunc, symbolTable); in runOnOperation() local
289 auto kernelModule = cast<gpu::GPUModuleOp>(Operation::create(state)); in createKernelModule() local
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.cc272 auto kernelModule = ::mlir::ModuleOp::create(module.getLoc()); in ExtractKernelModule() local
/external/llvm-project/mlir/lib/Dialect/GPU/IR/
DGPUDialect.cpp95 auto kernelModule = module.lookupSymbol<GPUModuleOp>(kernelModuleName); in verifyOperationAttribute() local
437 auto kernelModule = kernelFunc->getParentOfType<GPUModuleOp>(); in build() local
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertLaunchFuncToRuntimeCalls.cpp586 auto kernelModule = SymbolTable::lookupNearestSymbolFrom<gpu::GPUModuleOp>( in matchAndRewrite() local