Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dhlo_xla_runtime_pipeline.cc118 pm.addNestedPass<FuncOp>(tensorflow::CreateSymbolicShapeOptimizationPass( in CreateHloXlaRuntimePipeline()
137 pm.addNestedPass<FuncOp>(tensorflow::CreateSymbolicShapeOptimizationPass()); in CreateHloXlaRuntimePipeline()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_jitrt_pipeline.cc155 CreateSymbolicShapeOptimizationPass(/*constraints_only=*/true)); in CreateTfJitRtPipeline()
173 pm.addNestedPass<FuncOp>(CreateSymbolicShapeOptimizationPass()); in CreateTfJitRtPipeline()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_passes.h99 CreateSymbolicShapeOptimizationPass(bool constraints_only = false);
H A Dtf_jitrt_symbolic_shape_optimization.cc320 std::unique_ptr<OperationPass<FuncOp>> CreateSymbolicShapeOptimizationPass( in CreateSymbolicShapeOptimizationPass() function
H A Dtf_jitrt_passes.td172 let constructor = "tensorflow::CreateSymbolicShapeOptimizationPass()";