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

Handle grappler failure in savedmodel importer properly

Browse Source 
PiperOrigin-RevId: 351914138
Change-Id: I94db8b279385ea91483ef9f47bb43b7fe23180ed
This commit is contained in:
Kuangyuan Chen 2021-01-14 17:52:15 -08:00 committed by TensorFlower Gardener
parent 571c19440d
commit aebcab97df
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
View File

@ -3368,7 +3368,7 @@ class SavedModelSignatureDefImporterLite {
const GraphDebugInfo& debug_info() const { return debug_info_; }
private:
const MetaGraphDef& meta_graph_def_;
MetaGraphDef meta_graph_def_;
const GraphDebugInfo& debug_info_;
std::unique_ptr<Graph> graph_;
absl::Span<std::string> exported_names_;
@ -3379,10 +3379,11 @@ class SavedModelSignatureDefImporterLite {
Status SavedModelSignatureDefImporterLite::InitializeGraph(
MLIRImportOptions import_options) {
// TODO(jpienaar): Remove need to const_cast.
if (import_options.enable_grappler)
TF_RETURN_IF_ERROR(
RunGrappler(const_cast<MetaGraphDef*>(&meta_graph_def_)));
if (import_options.enable_grappler) {
// Grappler is best-effort.
auto status = RunGrappler(&meta_graph_def_);
if (!status.ok()) LOG(WARNING) << status;
}
GraphDef graph_def = meta_graph_def_.graph_def();
if (import_options.upgrade_legacy) {

4
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
View File

@ -130,6 +130,10 @@ Status RunGrappler(MetaGraphDef* meta_graph_def) {
std::unique_ptr<grappler::GrapplerItem> item =
grappler::GrapplerItemFromMetaGraphDef("graph", *meta_graph_def,
grappler::ItemConfig());
if (!item) {
return tensorflow::errors::Internal(
"Failed to create grappler item from MetaGraphDef.");
}
grappler::VirtualCluster cluster(&dev_set);
return grappler::RunMetaOptimizer(std::move(*item), config_proto, cpu_device,