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()
auto bcast_m =
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));
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)
......@@ -851,7 +851,7 @@ void ngraph::pass::CoreFusion::construct_conv_bias_add()
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);
}
......
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