Keras to ONNX model export for Barracuda in Unity3D

Question

this is my neural network model:

The input is an example of 10000 features. Each feature is a number (0 or 1). The output is a number between 0 and 1.

from tensorflow.keras.datasets import imdb
(train_data, train_labels), (test_data, test_labels) = imdb.load_data(
num_words=10000)
 
import numpy as np
def vectorize_sequences(sequences, dimension=10000):
  results = np.zeros((len(sequences), dimension))
  for i, sequence in enumerate(sequences):
    for j in sequence:
      results[i, j] = 1.
  return results
 
x_train = vectorize_sequences(train_data)
x_test = vectorize_sequences(test_data)
 
y_train = np.asarray(train_labels).astype("float32")
y_test = np.asarray(test_labels).astype("float32")
 
from tensorflow import keras
from tensorflow.keras import layers
 
model = keras.Sequential([
layers.Dense(16, activation="relu"),
layers.Dense(16, activation="relu"),
layers.Dense(1, activation="sigmoid")
])
 
model.compile(optimizer="rmsprop",
loss="binary_crossentropy",
metrics=["accuracy"])
 
x_val = x_train[:10000]
partial_x_train = x_train[10000:]
y_val = y_train[:10000]
partial_y_train = y_train[10000:]
 
model.fit(partial_x_train,
partial_y_train,
epochs=20,
batch_size=512,
validation_data=(x_val, y_val))

I exported the model in this way:

import tensorflow as tf
 
spec = (tf.TensorSpec(model.inputs[0].shape, tf.float32, name="my input"),)
 
nchw_inputs_list = [model.inputs[0].name]
 
import tf2onnx
 
model_proto, _ = tf2onnx.convert.from_keras(model, input_signature=spec, custom_ops=None, opset=9, inputs_as_nchw=nchw_inputs_list, output_path="example.onnx")

And when I import it in Unity:

Can someone please help me to export a simple model from Keras to ONNX and import it in the right way in Unity3D?

Thank you.