Publishing 2019 R1 content

[platform/upstream/dldt.git] / inference-engine / thirdparty / clDNN / src / max_unpooling.cpp

diff --git a/inference-engine/thirdparty/clDNN/src/max_unpooling.cpp b/inference-engine/thirdparty/clDNN/src/max_unpooling.cpp
index da67f02..5e2d99b 100644 (file)
--- a/inference-engine/thirdparty/clDNN/src/max_unpooling.cpp
+++ b/inference-engine/thirdparty/clDNN/src/max_unpooling.cpp
@@ -28,8 +28,16 @@ primitive_type_id max_unpooling_type_id()
     return &instance;
 }
 
+max_unpooling_node::typed_program_node(const std::shared_ptr<max_unpooling> prim, program_impl& prog) 
+    : parent(prim, prog)
+{
+    can_share_buffer(false); // for max_unpooling initial zero values are significant
+}
+
 layout max_unpooling_inst::calc_output_layout(max_unpooling_node const& node)
 {
+    assert((bool)node.get_primitive()->output_data_type == false
+          && "Output data type forcing is not supported for max_unpooling_node!");
     auto desc = node.get_primitive();
 
     auto input_layout = node.input().get_output_layout();