experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add dynamic_rnn into MLIR generated_examples test.

Browse Source 
PiperOrigin-RevId: 257243999
This commit is contained in:
Haoliang Zhang 2019-07-09 12:08:22 -07:00 committed by TensorFlower Gardener
parent b1899424ea
commit 1059951b8e
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
View File

@ -100,6 +100,16 @@ void AddTFToTFLConversionPasses(bool emit_builtin_tflite_ops, bool run_quantize,
bool lower_tensor_list_ops,
mlir::PassManager *pass_manager) {
pass_manager->addPass(mlir::TFControlFlow::CreateRaiseTFControlFlowPass());
if (lower_tensor_list_ops) {
// Execute this pass before `CanonicalizerPass` in case some TensorList
// ops are constant folded into variant types.
// TODO(haoliang): Add this pass by default.
// TODO(b/137038401): Handle other ops that operate on variant types and
// improve error reporting.
pass_manager->addPass(mlir::TFL::CreateLowerStaticTensorListPass());
}
// TODO(jpienaar): Revise post dialect constants.
pass_manager->addPass(mlir::TF::CreateDecodeConstantPass());
// Canonicalization includes const folding, which is utilized here to optimize
@ -112,10 +122,6 @@ void AddTFToTFLConversionPasses(bool emit_builtin_tflite_ops, bool run_quantize,
if (emit_builtin_tflite_ops) {
// Prepare for TFLite dialect, rerun canonicalization, and then legalize to
// the TFLite dialect.
// TODO(haoliang): Add this pass by default.
if (lower_tensor_list_ops) {
pass_manager->addPass(mlir::TFL::CreateLowerStaticTensorListPass());
}
pass_manager->addPass(mlir::TFL::CreatePrepareTFPass());
pass_manager->addPass(mlir::createCanonicalizerPass());
pass_manager->addPass(mlir::TFL::CreateLegalizeTFPass());