Я преобразовал свою модель h5 в tflite. Работает нормально с процессором. Но я не мог заставить его работать с делегатом TFLiteGPU. Это показывает, что «3 не может быть обработано этим делегатом. Только первые 0 операций будут выполняться на GPU, а остальные 30 на CPU».
Я использовал следующие уровни:
tf.compat.v1.keras.layers.Conv2D(16, kernel_size=(3, 3), strides=1, input_shape=input_shape,activation='relu')
tf.compat.v1.keras.layers.MaxPooling2D( pool_size=( 2 , 2 ),strides=2 )
tf.compat.v1.keras.layers.Dense( 4, activation='linear' )
Это не полная модель. Я только что добавил образцы слоев, которые я использовал.