Commit c915e05c authored by Anna Alberska's avatar Anna Alberska Committed by Robert Kimball

IntelGPU backend: AvgPoolBackprop operation (#1497)

* IntelGPU backend: AvgPoolBackprop operation

* Requested changes done

* Make gen_window_loop() static & initiate win_elems_size through shape_size
parent 8022982f
......@@ -514,6 +514,24 @@ bool runtime::intelgpu::IntelGPUBackend::compile(shared_ptr<Function> func)
avg_pool->get_padding_below(),
mode);
}
else if ("AvgPoolBackprop" == op->description())
{
arguments_check(op, 1, 1);
const shared_ptr<op::AvgPoolBackprop> avg_pool_b =
static_pointer_cast<op::AvgPoolBackprop>(op);
do_avg_pool_backprop_operation(topology,
get_input_name(op, 0),
get_input_shape(op, 0),
get_output_name(op),
get_output_shape(op),
get_output_type(op),
avg_pool_b->get_window_shape(),
avg_pool_b->get_window_movement_strides(),
avg_pool_b->get_padding_below(),
avg_pool_b->get_include_padding_in_avg_computation());
}
else if ("Broadcast" == op->description())
{
arguments_check(op, 1, 1);
......
......@@ -54,6 +54,17 @@ namespace ngraph
const Shape& win_stride,
const Shape& pad_below);
void do_avg_pool_backprop_operation(cldnn::topology& topology,
const std::string& delta_name,
const Shape& delta_shape,
const std::string& output_name,
const Shape& output_shape,
const element::Type& output_type,
const Shape& win_shape,
const Shape& win_stride,
const Shape& pad_below,
const bool include_padding);
void do_dot_operation(cldnn::topology& topology,
const std::string& inputA_name,
const Shape& inputA_shape,
......
......@@ -7,10 +7,7 @@ backwards_abs
backwards_atan
backwards_avgpool_n1_c1_hw2x2
backwards_avgpool_n1_c1_hw4x4
backwards_avgpool_n2_c2_hw2x2_win_2x2_str_1x1_padding_numeric
backwards_avgpool_n2_c2_hw4x4
backwards_avgpool_n2_c2_hw4x4_numeric
backwards_avgpool_n2_c2_hw4x4_win_2x2_str_1x1_numeric
backwards_batch_norm_three_outputs
backwards_ceiling
backwards_dot_scalar_tensor
......
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