在tensorflow中添加一个op,但是类型是错误的

时间:2018-11-26 01:44:59

标签: tensorflow-lite

我需要在tensorflow和lite中添加一个OP。代码大约是这样的:

REGISTER_OP("CheckSize") .Input("y: uint8")
  • 如果类型为int32,则一切正常(生成我的模型并成功转移到lite模型)。
  • 但是如果输入类型是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输入类型?

0 个答案:

没有答案