commit bf640a47a03143796a8e0fa3e04b806460ddb944
author Kuangyuan Chen <chky@google.com> Fri Jun 19 15:51:19 2020 -0700
committer Copybara-Service <copybara-worker@google.com> Fri Jun 19 15:53:28 2020 -0700
tree 722f6ac3a9d04b9f44033810b151850fc19a475f
parent 6d4ab1d60278a67a804a3f6288c27e590e4c27aa

Import initialization graph in SignatureDef SavedModels as an MLIR function in
TF saved model dialect.
-Mark the init function referenced by SessionInitializerOp as an exported
function with the reserved name "__tf_saved_model_session_initializer".
-Remove variable init logic in init function if it is already imported as a global tensor.
-Add a canonicalizer to SessionInitializerOp to remove empty init functions.
-Return error on SessionInitializerOp in IREE compiler pipeline.

PiperOrigin-RevId: 317395165

integrations/tensorflow/compiler/TFSavedModelLowerGlobalTensors.cpp

diff --git a/integrations/tensorflow/compiler/TFSavedModelLowerGlobalTensors.cpp b/integrations/tensorflow/compiler/TFSavedModelLowerGlobalTensors.cpp
index 0cca1b1..6d8271d 100644
--- a/integrations/tensorflow/compiler/TFSavedModelLowerGlobalTensors.cpp
+++ b/integrations/tensorflow/compiler/TFSavedModelLowerGlobalTensors.cpp
@@ -60,6 +60,10 @@
   OpBuilder globalBuilder(module.getBodyRegion());
   SymbolTable symbolTable(module);
 
+  if (auto sessionInitializer = tf_saved_model::GetSessionInitializerOp(module))
+    return sessionInitializer.emitError()
+           << "Session initializer is not supported yet";
+
   DenseMap<StringRef, std::string> symNameToFlowSymName;
   for (auto globalTensor : module.getOps<tf_saved_model::GlobalTensorOp>()) {
     auto exportedNames = tf_saved_model::GetExportedNames(globalTensor);