Add the fp16 support for IsNaN input

Fixed the heap-buffer-overflow exception

Author: sgbihu

src/tests/functional/base_func_tests/src/base/utils/generate_inputs.cpp

@@ -843,16 +843,6 @@ ov::Tensor generate(const std::shared_ptr<ov::op::v10::IsFinite>& node,
     return tensor;
 }
 
-ov::Tensor generate(const std::shared_ptr<ov::op::v10::IsNaN>& node,
-                    size_t port,
-                    const ov::element::Type& elemType,
-                    const ov::Shape& targetShape,
-                    std::shared_ptr<InputGenerateData> inGenRangeData = nullptr) {
-    ov::Tensor tensor{elemType, targetShape};
-    comparison::fill_tensor(tensor);
-    return tensor;
-}
-
 namespace is_inf {
 template <typename T>
 void fill_tensor(ov::Tensor& tensor) {
@@ -877,6 +867,20 @@ void fill_tensor(ov::Tensor& tensor) {
 }
 } // namespace is_inf
 
+ov::Tensor generate(const std::shared_ptr<ov::op::v10::IsNaN>& node,
+                    size_t port,
+                    const ov::element::Type& elemType,
+                    const ov::Shape& targetShape,
+                    std::shared_ptr<InputGenerateData> inGenRangeData = nullptr) {
+    ov::Tensor tensor{elemType, targetShape};
+    if (elemType == ov::element::f16) {
+        is_inf::fill_tensor<ov::float16>(tensor);
+    } else {
+        is_inf::fill_tensor<float>(tensor);
+    }
+    return tensor;
+}
+
 ov::Tensor generate(const std::shared_ptr<ov::op::v10::IsInf>& node,
                     size_t port,
                     const ov::element::Type& elemType,