Publishing 2019 R1 content

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

diff --git a/inference-engine/thirdparty/clDNN/src/apply_adam.cpp b/inference-engine/thirdparty/clDNN/src/apply_adam.cpp
index 1b0e9d4..24b659e 100644 (file)
--- a/inference-engine/thirdparty/clDNN/src/apply_adam.cpp
+++ b/inference-engine/thirdparty/clDNN/src/apply_adam.cpp
@@ -27,8 +27,15 @@ primitive_type_id apply_adam_type_id()
     return &instance;
 }
 
+apply_adam_node::typed_program_node(const std::shared_ptr<apply_adam> prim, program_impl& prog)
+    : parent(prim, prog)
+{
+    can_share_buffer(false); //apply adam's output initial val should be either 0 or use same buffer as mutable_data after it (no allocation needed)
+}
 layout apply_adam_inst::calc_output_layout(apply_adam_node const& node)
 {
+    assert((bool)node.get_primitive()->output_data_type == false
+           && "Output data type forcing is not supported for apply_adam_node!");
     return node.input().get_non_padded_output_layout();
 }
 
@@ -72,4 +79,4 @@ apply_adam_inst::typed_primitive_inst(network_impl& network, apply_adam_node con
     CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "beta1_power format", beta1_power_format.value, "supported beta1_power formats", format::yxfb, format::bfyx);
     CLDNN_ERROR_NOT_PROPER_FORMAT(node.id(), "beta2_power format", beta2_power_format.value, "supported beta2_power formats", format::yxfb, format::bfyx);
 }
-}
\ No newline at end of file
+}