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

Refine TODO items related to lower-static-tensor-list pass.

Browse Source 
PiperOrigin-RevId: 257274127
This commit is contained in:
Haoliang Zhang 2019-07-09 14:40:01 -07:00 committed by TensorFlower Gardener
parent f5ec649ad6
commit 83788ea45a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -104,9 +104,9 @@ void AddTFToTFLConversionPasses(bool emit_builtin_tflite_ops, bool run_quantize,
if (lower_tensor_list_ops) {
// Execute this pass before `CanonicalizerPass` in case some TensorList
// ops are constant folded into variant types.
// TODO(b/137125056): Move this pass after `CanonicalizerPass` after we
// handle constant ops that produce `TensorList`.
// 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());
}