Publishing 2019 R1 content

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

diff --git a/inference-engine/thirdparty/clDNN/src/softmax_loss_grad.cpp b/inference-engine/thirdparty/clDNN/src/softmax_loss_grad.cpp
index df94b60..41069f5 100644 (file)
--- a/inference-engine/thirdparty/clDNN/src/softmax_loss_grad.cpp
+++ b/inference-engine/thirdparty/clDNN/src/softmax_loss_grad.cpp
@@ -28,6 +28,9 @@ primitive_type_id softmax_loss_grad_type_id()
 
 layout softmax_loss_grad_inst::calc_output_layout(softmax_loss_grad_node const& node)
 {
+    assert((bool)node.get_primitive()->output_data_type == false
+           && "Output data type forcing is not supported for "
+              "softmax_loss_grad_node!");
     return node.input().get_non_padded_output_layout();
 }