diff --git a/tensorflow/core/grappler/optimizers/constant_folding.cc b/tensorflow/core/grappler/optimizers/constant_folding.cc index 09a4716df51b48..7cf26afb67299c 100644 --- a/tensorflow/core/grappler/optimizers/constant_folding.cc +++ b/tensorflow/core/grappler/optimizers/constant_folding.cc @@ -102,7 +102,7 @@ string AsControlDependency(const NodeDef& node) { ConstantFolding::ConstantFolding() { ops_to_preserve_ = std::regex( "Placeholder.*|Const|.*Save.*|.*Restore.*|.*Reader|Enter|Exit|" - "NextIteration"); + "NextIteration|.*Quantized.*"); } string ConstantFolding::AddControlDependency(const string& input_name) {