fix type mismatch (#4077)

src/ngraph/op/fused/softmax_crossentropy.cpp

@@ -231,7 +231,7 @@ NodeVector op::SoftmaxCrossEntropyBackprop::decompose_op() const
         auto mask_constant =
             ngraph::op::Constant::create(element::i64, labels.get_shape(), {m_ignore_index});
         auto not_equal = std::make_shared<ngraph::op::NotEqual>(labels, mask_constant);
-        auto convert = std::make_shared<ngraph::op::Convert>(not_equal, element::f64);
+        auto convert = std::make_shared<ngraph::op::Convert>(not_equal, delta.get_element_type());
         auto reshape = std::make_shared<ngraph::op::Reshape>(
             convert, AxisVector{0, 1}, Shape{convert->get_shape().at(0)});
         auto broadcast_mask =
@@ -242,7 +242,8 @@ NodeVector op::SoftmaxCrossEntropyBackprop::decompose_op() const
             make_shared<op::Reshape>(labels, AxisVector{0, 1}, Shape{labels.get_shape().at(0)});
         auto one_hot =
             std::make_shared<ngraph::op::OneHot>(reshape_labels, softmax.get_shape(), one_hot_axis);
-        auto convert_one_hot = std::make_shared<ngraph::op::Convert>(one_hot, element::f64);
+        auto convert_one_hot =
+            std::make_shared<ngraph::op::Convert>(one_hot, delta.get_element_type());
 
         if (delta.get_shape() != convert_one_hot->get_shape())
         {