TensorRT 不支持Tensorflow的操作有如下

tf.unpack, tf.slice, tf.tile, tf.expand_dims, tf.fill, tf.cast, tf.floor_div, tf.range

比较坑,所以你必须限制你的模型使用的方法,尽量选择简便的操作方式

posted @ 2019-03-19 19:16  今夜无风  阅读(2559)  评论(0编辑  收藏  举报