Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dpasses.h97 CreateGpuKernelToRocdlPass();
Dkernel_lowering_passes.cc97 std::unique_ptr<OperationPass<GPUModuleOp> > CreateGpuKernelToRocdlPass() { in CreateGpuKernelToRocdlPass() function
Dpasses.td71 let constructor = "transforms::CreateGpuKernelToRocdlPass()";
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc364 kernelPm.addPass(mlir::kernel_gen::transforms::CreateGpuKernelToRocdlPass()); in LowerKernelBodiesToLowLevelIr()