Searched defs:launchOp (Results 1 – 7 of 7) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | KernelOutlining.cpp | 106 LogicalResult mlir::sinkOperationsIntoLaunchOp(gpu::LaunchOp launchOp) { in sinkOperationsIntoLaunchOp()
138 static gpu::GPUFuncOp outlineKernelFuncImpl(gpu::LaunchOp launchOp, in outlineKernelFuncImpl()
197 gpu::GPUFuncOp mlir::outlineKernelFunc(gpu::LaunchOp launchOp, in outlineKernelFunc()
214 static void convertToLaunchFuncOp(gpu::LaunchOp launchOp, in convertToLaunchFuncOp()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToVulkan/ |
| D | ConvertGPULaunchFuncToVulkanLaunchFunc.cpp | 101 Location loc, gpu::LaunchFuncOp launchOp) { in declareVulkanLaunchFunc()
150 gpu::LaunchFuncOp launchOp) { in convertGpuLaunchFunc()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 218 auto launchOp = builder.create<gpu::LaunchOp>( in createLaunch() local
378 ParallelOp parallelOp, gpu::LaunchOp launchOp, in processParallelLoop()
577 gpu::LaunchOp launchOp = rewriter.create<gpu::LaunchOp>( in matchAndRewrite() local
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 485 gpu::LaunchFuncOp launchOp, ArrayRef<Value> operands, in generateParamsArray()
566 gpu::LaunchFuncOp launchOp, ArrayRef<Value> operands, in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| D | ConvertLaunchFuncToLLVMCalls.cpp | 154 matchAndRewrite(gpu::LaunchFuncOp launchOp, ArrayRef<Value> operands, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | passes.cc | 211 getFunction().walk([&](mlir::gpu::LaunchFuncOp launchOp) { in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 80 auto walkResult = module.walk([&module](LaunchFuncOp launchOp) -> WalkResult { in verifyOperationAttribute()
|