diff options
author | 2018-01-23 12:24:21 -0800 | |
---|---|---|
committer | 2018-01-23 12:28:02 -0800 | |
commit | ba4aec48268d02f111cd7e2c2666f4e7b077e68a (patch) | |
tree | b5b2e456b9499cc460381bdb471e864b8b889dfa /tensorflow/contrib/lite/toco/graph_transformations/quantize.cc | |
parent | b47119257b76bdec542121aaed056c8b010fd19f (diff) |
Internal Change
PiperOrigin-RevId: 182974191
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/quantize.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/quantize.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/quantize.cc b/tensorflow/contrib/lite/toco/graph_transformations/quantize.cc index 56082b965a..b973b2b813 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/quantize.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/quantize.cc @@ -412,7 +412,7 @@ bool Quantize::Run(Model* model, std::size_t op_index) { model->flags.set_output_arrays(i, dequantize_op->inputs[0]); } } - model->arrays.erase(dequantize_op->outputs[0]); + model->EraseArray(dequantize_op->outputs[0]); model->operators.erase(dequantize_it); } } |