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.h84 std::unique_ptr<mlir::FunctionPass> CreateMapParallelLoopsPass();
Dpasses.td114 let constructor = "transforms::CreateMapParallelLoopsPass()";
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dkernel_creator.cc267 mlir::kernel_gen::transforms::CreateMapParallelLoopsPass()); in LowerLoopsToGPUorCPU()