Skip to content

N
ngraph
Commit 970f5006 authored by Dmitry Yershov's avatar Dmitry Yershov Committed by Robert Kimball
Browse files
Options

IntelGPU backend: More cases of MaxPool op to be executed via clDNN (#2938)

parent 1cbc1517
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
src/ngraph/runtime/intelgpu/intelgpu_backend.cpp
View file @ 970f5006
......@@ -874,9 +874,7 @@ shared_ptr<runtime::Executable>
const shared_ptr<op::MaxPool> max_pool = static_pointer_cast<op::MaxPool>(op);
if ((op->get_input_shape(0).size() > 4) ||
(op->get_output_element_type(0) != element::f32) ||
has_non_zero(max_pool->get_padding_below()) ||
has_non_zero(max_pool->get_padding_above()))
(op->get_output_element_type(0) != element::f32))
{
const shared_ptr<Node> def_val = max_pool->get_default_value();
const shared_ptr<op::Constant> def_const =
......
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