Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dmap_parallel_loops_to_gpu.cc35 std::unique_ptr<mlir::FunctionPass> CreateMapParallelLoopsPass() { in CreateMapParallelLoopsPass() function
Dpasses.h95 std::unique_ptr<mlir::FunctionPass> CreateMapParallelLoopsPass();
Dpasses.td141 let constructor = "transforms::CreateMapParallelLoopsPass()";
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc344 mlir::kernel_gen::transforms::CreateMapParallelLoopsPass()); in LowerLoopsToGPUorCPU()