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

Fix the mnist test for keras quantization api

Browse Source 
PiperOrigin-RevId: 286462510
Change-Id: I6bb71d1d9151cddf04e3ceb6c32ec22039c2ae7e
This commit is contained in:
Feng Liu 2019-12-19 14:19:43 -08:00 committed by TensorFlower Gardener
parent 0535f25cf0
commit 59166604c2
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc
View File

@ -277,7 +277,6 @@ Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags,
auto status = ConvertTFExecutorToTFLOrFlatbuffer( auto status = ConvertTFExecutorToTFLOrFlatbuffer(
module.get(), /*export_to_mlir=*/false, emit_builtin_tflite_ops, module.get(), /*export_to_mlir=*/false, emit_builtin_tflite_ops,
emit_select_tf_ops, emit_custom_ops, quant_specs, result, &pm); emit_select_tf_ops, emit_custom_ops, quant_specs, result, &pm);
if (toco_flags.has_dump_graphviz_dir()) { if (toco_flags.has_dump_graphviz_dir()) {
TF_RETURN_IF_ERROR(DumpOpGraphToFile( TF_RETURN_IF_ERROR(DumpOpGraphToFile(
// rename once we enable the new converter feature flag. // rename once we enable the new converter feature flag.