Merge pull request #1070 from dkurt:tf_half_pixel_for_nn

Test data for half pixel NN resize from ONNX

Author: asmorkalov

testdata/dnn/onnx/data/input_tf_half_pixel_for_nn.npy

@@ -2451,7 +2451,7 @@ def save_data_and_tf_function(tf_function, name, input):
     np.save(os.path.join("data", "output_" + name + ".npy"), output)
     cumsum_model = tf2onnx.convert.from_function(
         function=tf_function,
-        input_signature=[tf.TensorSpec([], tf.float32)],
+        input_signature=[tf.TensorSpec(input.shape, tf.float32)],
         opset=14)[0]
     onnx.save(cumsum_model, os.path.join("models", name + ".onnx"))
 
@@ -2977,3 +2977,9 @@ def make_node_with_constant(op_type, constant_value, inputs=None, outputs=None,
 np.save(input_files, input.data)
 output_files = os.path.join("data", "output_" + name)
 np.save(output_files, np.ascontiguousarray(output.data))
+
+## tf_half_pixel_for_nn
+@tf.function
+def tf_resize_nearest(x):
+    return tf.compat.v1.image.resize_nearest_neighbor(x, size=(5, 6), align_corners=False, half_pixel_centers=True)
+save_data_and_tf_function(tf_resize_nearest, "tf_half_pixel_for_nn", np.random.rand(1, 2, 3, 2))