Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_rewrite_vector_multi_reduction.cc79 createRewriteVectorMultiReductionPass() { in createRewriteVectorMultiReductionPass() function
Dtf_jitrt_passes.h73 createRewriteVectorMultiReductionPass();
Dtf_jitrt_passes.td240 let constructor = "tensorflow::createRewriteVectorMultiReductionPass()";
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dhlo_xla_runtime_pipeline.cc216 pm.addNestedPass<FuncOp>(tensorflow::createRewriteVectorMultiReductionPass()); in CreateHloXlaRuntimePipeline()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_jitrt_pipeline.cc251 pm.addNestedPass<FuncOp>(createRewriteVectorMultiReductionPass()); in CreateTfJitRtPipeline()