From 59166604c26b8ffde742389fcd99c8090bf8ec04 Mon Sep 17 00:00:00 2001 From: Feng Liu Date: Thu, 19 Dec 2019 14:19:43 -0800 Subject: [PATCH] Fix the mnist test for keras quantization api PiperOrigin-RevId: 286462510 Change-Id: I6bb71d1d9151cddf04e3ceb6c32ec22039c2ae7e --- .../compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc | 1 - 1 file changed, 1 deletion(-) diff --git a/tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc b/tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc index fcd2c38b37a..7738f1eb037 100644 --- a/tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc +++ b/tensorflow/compiler/mlir/lite/python/graphdef_to_tfl_flatbuffer.cc @@ -277,7 +277,6 @@ Status ConvertGraphDefToTFLiteFlatBuffer(const toco::ModelFlags& model_flags, auto status = ConvertTFExecutorToTFLOrFlatbuffer( module.get(), /*export_to_mlir=*/false, emit_builtin_tflite_ops, emit_select_tf_ops, emit_custom_ops, quant_specs, result, &pm); - if (toco_flags.has_dump_graphviz_dir()) { TF_RETURN_IF_ERROR(DumpOpGraphToFile( // rename once we enable the new converter feature flag.