Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/
H A Dgpu_passes.h41 createGpuKernelToNvvmPass();
H A Dgpu_passes.td29 let constructor = "createGpuKernelToNvvmPass()";
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dhlo_to_gpu_pipeline.cc73 pm.addNestedPass<GPUModuleOp>(createGpuKernelToNvvmPass()); in createHloToGpuPipeline()
H A Dgpu_kernel_lowering_passes.cc92 mlir::createGpuKernelToNvvmPass() { in createGpuKernelToNvvmPass() function in mlir
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc339 kernelPm.addPass(mlir::createGpuKernelToNvvmPass()); in LowerKernelBodiesToLowLevelIr()