Commit f054faaf authored by Nishant Patel's avatar Nishant Patel Committed by Scott Cyphers

Change shared_ptr<Node> to Output<Node> for AvgPoolBackprop (#3754)

parent 18d20b7b
......@@ -234,13 +234,13 @@ shared_ptr<Node> op::v0::AvgPool::get_default_value() const
}
op::v0::AvgPoolBackprop::AvgPoolBackprop(const Shape& forward_arg_shape,
const shared_ptr<Node>& delta,
const Output<Node>& delta,
const Shape& window_shape,
const Strides& window_movement_strides,
const Shape& padding_below,
const Shape& padding_above,
bool include_padding_in_avg_computation)
: Op(check_single_output_args({delta}))
: Op({delta})
, m_forward_arg_shape(forward_arg_shape)
, m_window_shape(window_shape)
, m_window_movement_strides(window_movement_strides)
......
......@@ -179,7 +179,7 @@ namespace ngraph
const NodeTypeInfo& get_type_info() const override { return type_info; }
AvgPoolBackprop() = default;
AvgPoolBackprop(const Shape& forward_arg_shape,
const std::shared_ptr<Node>& delta,
const Output<Node>& delta,
const Shape& window_shape,
const Strides& window_movement_strides,
const Shape& padding_below,
......
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