Home
last modified time | relevance | path

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

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