From 33c3aeb867f74ceae64312be02d5a53193a51a01 Mon Sep 17 00:00:00 2001 From: Lukasz Debski Date: Sat, 5 Sep 2020 18:01:43 +0200 Subject: [PATCH] [IE CLDNN] Fixing blocked format opting for strided_slice (#2073) --- inference-engine/thirdparty/clDNN/src/program.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/inference-engine/thirdparty/clDNN/src/program.cpp b/inference-engine/thirdparty/clDNN/src/program.cpp index 0a0e596..2030418 100644 --- a/inference-engine/thirdparty/clDNN/src/program.cpp +++ b/inference-engine/thirdparty/clDNN/src/program.cpp @@ -61,6 +61,7 @@ #include "split_inst.h" #include "mvn_inst.h" #include "reduce_inst.h" +#include "strided_slice_inst.h" #include "to_string_utils.h" #include "gpu/memory_gpu.h" @@ -1168,7 +1169,8 @@ void program_impl::set_layout_optimizer_attributes(layout_optimizer& lo) { || prim.as().get_primitive()->across_channels) && prim.type() != cldnn::arg_max_min::type_id() && prim.type() != cldnn::mutable_data::type_id() && - prim.type() != cldnn::reduce::type_id()) + prim.type() != cldnn::reduce::type_id() && + prim.type() != cldnn::strided_slice::type_id()) can_use_fsv16 = false; if (prim.type() == cldnn::quantize::type_id() && -- 2.7.4