Skip to content

N
ngraph
Commit 6562c902 authored by Jayaram Bobba's avatar Jayaram Bobba Committed by Scott Cyphers
Browse files
Options

static_cast where possible (#2850)

parent 870f5000
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/ngraph/pass/core_fusion.cpp
View file @ 6562c902
...@@ -772,11 +772,11 @@ void ngraph::pass::CoreFusion::construct_conv_bias() ...@@ -772,11 +772,11 @@ void ngraph::pass::CoreFusion::construct_conv_bias()
auto bcast_m = auto bcast_m =
std::dynamic_pointer_cast<op::Broadcast>(m.get_match_root()->get_argument(1)); std::dynamic_pointer_cast<op::Broadcast>(m.get_match_root()->get_argument(1));
if (conv_m == nullptr) if (conv_m == nullptr || bcast_m == nullptr)
{ {
conv_m = std::dynamic_pointer_cast<ngraph::op::Convolution>( conv_m = std::static_pointer_cast<ngraph::op::Convolution>(
m.get_match_root()->get_argument(1)); m.get_match_root()->get_argument(1));
bcast_m = std::dynamic_pointer_cast<op::Broadcast>(m.get_match_root()->get_argument(0)); bcast_m = std::static_pointer_cast<op::Broadcast>(m.get_match_root()->get_argument(0));
} }
if (conv_m->get_shape().size() > 5 || conv_m->get_element_type() != element::f32) if (conv_m->get_shape().size() > 5 || conv_m->get_element_type() != element::f32)
...@@ -851,7 +851,7 @@ void ngraph::pass::CoreFusion::construct_conv_bias_add() ...@@ -851,7 +851,7 @@ void ngraph::pass::CoreFusion::construct_conv_bias_add()
if (!conv_m) if (!conv_m)
{ {
conv_m = std::dynamic_pointer_cast<ngraph::op::ConvolutionBias>(add_m->get_argument(0)); conv_m = std::static_pointer_cast<ngraph::op::ConvolutionBias>(add_m->get_argument(0));
add_input_m = add_m->get_argument(1); add_input_m = add_m->get_argument(1);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment