From 9c4a57e69bae2986c34fa919339fa821b77f77b8 Mon Sep 17 00:00:00 2001 From: Karim Nosir Date: Mon, 5 Oct 2020 16:12:24 -0700 Subject: [PATCH] Disable printing the ops on error in TFLite converter. PiperOrigin-RevId: 335521410 Change-Id: Ic54a57683ee796eb19116c3d8b4f861f22e1356b --- tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc b/tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc index c3fc2bcdaaf..a2387e89483 100644 --- a/tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc +++ b/tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc @@ -139,6 +139,9 @@ Status ConvertTFExecutorToTFLOrFlatbuffer( bool emit_select_tf_ops, bool emit_custom_ops, const mlir::TFL::QuantizationSpecs& quant_specs, std::string* result, mlir::PassManager* pass_manager) { + // Explicitly disable dumping Op details on failures. + module.getContext()->printOpOnDiagnostic(false); + // Register a warning handler only log to std out. mlir::ScopedDiagnosticHandler s( module.getContext(), [](mlir::Diagnostic& diag) {