Home
last modified time | relevance | path

Searched refs:layout_optimization_options (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dtf_tfl_passes.cc289 mlir::TF::LayoutOptimizationPipelineOptions layout_optimization_options; in AddPostVariableFreezingTFToTFLConversionPasses() local
290 layout_optimization_options.force_data_format = kTFLiteDataLayout; in AddPostVariableFreezingTFToTFLConversionPasses()
291 layout_optimization_options.skip_fold_transpose_in_ops = true; in AddPostVariableFreezingTFToTFLConversionPasses()
293 pass_manager->nest<mlir::func::FuncOp>(), layout_optimization_options); in AddPostVariableFreezingTFToTFLConversionPasses()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dgraph_optimization_pass.cc58 LayoutOptimizationPipelineOptions layout_optimization_options; in Run() local
60 layout_optimization_options); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dtf_to_tfrt.cc2282 mlir::TF::LayoutOptimizationPipelineOptions layout_optimization_options; in CreateTFExecutorToTFPipeline() local
2283 layout_optimization_options.force_data_format = in CreateTFExecutorToTFPipeline()
2289 layout_optimization_options.skip_fold_transpose_in_ops = true; in CreateTFExecutorToTFPipeline()
2291 layout_optimization_options); in CreateTFExecutorToTFPipeline()