commit 3ccebbdbaeff9b65e4204c87825e71de4dd8d739
author Jaesung Chung <jaesung@google.com> Mon Jun 14 19:07:23 2021 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Jun 14 19:19:55 2021 -0700
tree 8f8c496a21caad62691f80b8871b06bc500a890e
parent 682b1adf8961ae14c2e25b82da4631d1288ce3ab

Remove output number restriction in flatbuffer importer

CallOnce op and other resource operators can have zero output.

PiperOrigin-RevId: 379396505
Change-Id: I450677fc4174558adde474b09e346a8f2a67dea9

tensorflow/compiler/mlir/lite/flatbuffer_import.cc

diff --git a/tensorflow/compiler/mlir/lite/flatbuffer_import.cc b/tensorflow/compiler/mlir/lite/flatbuffer_import.cc
index fe8f88c..f06170f 100644
--- a/tensorflow/compiler/mlir/lite/flatbuffer_import.cc
+++ b/tensorflow/compiler/mlir/lite/flatbuffer_import.cc
@@ -653,11 +653,6 @@
   llvm::SmallVector<Value, 4> operands;
   llvm::SmallVector<mlir::Type, 2> outputTypes;
 
-  if (op.outputs.empty()) {
-    auto err = errors::InvalidArgument("operator with no outputs");
-    return emitError(loc, err.ToString()), err;
-  }
-
   const tflite::OperatorCodeT& op_code = *op_codes.at(op.opcode_index);
 
   TF_ASSIGN_OR_RETURN(const std::string op_name, GetMlirOpName(op, op_code));