1. 系统信息

• 操作系统平台和发行版：Linux Ubuntu 22.04
• TensorFlow安装(pip包或从源代码构建):pip包
• TensorFlow库(版本，如果是pip包或github SHA,如果是从源代码构建):2.16.1

2. 代码

import tensorflow as tf
import pdb
print("TensorFlow version:", tf.__version__)

class network(tf.keras.Model):
    def __init__(self):
        super().__init__()
        self.conv1 = tf.keras.layers.Conv2D(filters=2, kernel_size=3, strides=2, activation='relu')

    def call(self, x):
        x = self.conv1(x)

        return x
    
model = network()

inp  = tf.keras.Input(shape=(28, 28, 1))
out = model(inp)
model.build(input_shape=(None, 28, 28, 1))

model.summary()

converter = tf.lite.TFLiteConverter.from_keras_model(model)
tfliteModel = converter.convert()

输出

LLVM ERROR: Failed to infer result type(s).
Aborted (core dumped

ouptut.txt