Home
last modified time | relevance | path

Searched refs:CreateTFShapeInferencePass (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbridge.cc99 pm.addPass(TF::CreateTFShapeInferencePass()); in CreateTPUBridgePipeline()
115 pm.addPass(TF::CreateTFShapeInferencePass()); in CreateTPUBridgePipeline()
143 pm.addPass(TF::CreateTFShapeInferencePass()); in CreateTPUBridgePipelineV1()
Dshape_inference_pass.cc41 std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass() { in CreateTFShapeInferencePass() function
Dgraph_optimization_pass.cc50 pm.addPass(CreateTFShapeInferencePass()); in Run()
Doptimize.cc163 pm.addPass(CreateTFShapeInferencePass()); in CreateTFStandardPipeline()
Dpasses.h65 std::unique_ptr<OperationPass<ModuleOp>> CreateTFShapeInferencePass();
Dtf_passes.td24 let constructor = "TF::CreateTFShapeInferencePass()";
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_passes.cc79 pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass()); in AddTFToTFLConversionPasses()
144 pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass()); in AddTFToTFLConversionPasses()
183 pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass()); in AddTFToTFLConversionPasses()
204 pass_manager->addPass(mlir::TF::CreateTFShapeInferencePass()); in AddTFToTFLConversionPasses()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc292 pm.addPass(mlir::TF::CreateTFShapeInferencePass()); in CreateConvertMlirToXlaHloPipeline()
300 pm.addPass(mlir::TF::CreateTFShapeInferencePass()); in CreateConvertMlirToXlaHloPipeline()
323 pm.addPass(mlir::TF::CreateTFShapeInferencePass()); in CreateConvertMlirToXlaHloPipeline()