Home
last modified time | relevance | path

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

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