Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dcollapse_parallel_loops_to_1d_pass.cc56 std::unique_ptr<OperationPass<>> mlir::createCollapseParallelLoopsTo1DPass() { in createCollapseParallelLoopsTo1DPass() function in mlir
H A Dhlo_to_gpu_pipeline.cc55 pm.addNestedPass<FuncOp>(createCollapseParallelLoopsTo1DPass()); in createHloToGpuPipeline()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Transforms/
H A Dpasses.h99 std::unique_ptr<OperationPass<>> createCollapseParallelLoopsTo1DPass();
H A Dpasses.td69 let constructor = "createCollapseParallelLoopsTo1DPass()";
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
H A Dkernel_creator.cc223 pm.addNestedPass<FuncOp>(mlir::createCollapseParallelLoopsTo1DPass()); in LowerTFtoLoops()