fix KW issue

Intel-tensorflow · Feb 4, 2020 · 7e3d47e · 7e3d47e
1 parent 5d80e1e
commit 7e3d47e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow/core/grappler/optimizers/remapper.cc b/tensorflow/core/grappler/optimizers/remapper.cc
@@ -567,6 +567,7 @@ bool FindContractionWithBiasInPort(const RemapperContext& ctx,
   if (add_node_view.NumRegularFanins() < port_id + 1) return false;
   const auto& bias_add_node_view =
       add_node_view.GetRegularFanin(port_id).node_view();
+  if (!bias_add_node_view) return false;
   const auto* bias_add_node_def = bias_add_node_view->node();
 
   if (!FindContractionWithBias(ctx, bias_add_node_view->node_index(), base,
@@ -626,6 +627,7 @@ bool FindContractionWithBiasAndAddActivation(
     const RemapperContext& ctx, int node_index,
     ContractionWithBiasAndAddActivation* matched) {
   const auto* node_view = ctx.graph_view.GetNode(node_index);
+  if (!node_view) return false;
   // TODO(lyandy): Forward controls for patterns with control dependencies.
   if (HasControlFaninOrFanout(*node_view)) return false;