标签: tensorflow size tensorflow-lite custom-operator
我正在尝试创建一个包含tf.Size的简单tensorflow模型,以便创建自己的自定义操作,这样我就不会使用Select TF ops feature并增加文件的大小
tf.Size
Select TF ops feature
但是在转换模型之后,将张量的大小添加到随后的任何操作中。即使我不使用转换器的标志allow_custom_ops,也会发生这种情况,据我了解,toco不会将其识别为自定义操作。
allow_custom_ops
有什么主意吗?