我需要在tensorflow和lite中添加一个OP。代码大约是这样的:
REGISTER_OP("CheckSize") .Input("y: uint8")
但是如果输入类型是uint8或int8(也许其他类型相同),则可以生成张量流模型,但是当转换为lite模型时,错误信息如下:
2018-11-26 09:25:12.366596:我 tensorflow / contrib / lite / toco / import_tensorflow.cc:1055]转换 不支持的操作:CheckSize 2018-11-26 09:25:12.366683:I tensorflow / contrib / lite / toco / import_tensorflow.cc:1055]转换 不支持的操作:CheckSize 2018-11-26 09:25:12.366702:I tensorflow / contrib / lite / toco / import_tensorflow.cc:1055]转换 不支持的操作:CheckSize 2018-11-26 09:25:12.366737:I tensorflow / contrib / lite / toco / import_tensorflow.cc:1055]转换 不支持的操作:CheckSize 2018-11-26 09:25:12.368419:F tensorflow / contrib / lite / toco / tooling_util.cc:967]检查失败: array-> has_shape()
任何人都知道原因,如何使用int8输入类型?