diff --git a/tensorflow/core/grappler/optimizers/remapper.cc b/tensorflow/core/grappler/optimizers/remapper.cc index eeeaac3fe3f13e..04c424c3c7a0df 100644 --- 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;