TensorFlow函数教程:tf.lite.toco_convert

2019-04-03 17:40 更新

tf.lite.toco_convert函数

别名:

  • tf.contrib.lite.toco_convert
  • tf.lite.toco_convert
tf.lite.toco_convert(
    input_data,
    input_tensors,
    output_tensors,
    *args,
    **kwargs
)

定义在:tensorflow/lite/python/convert.py

使用TOCO转换模型。(弃用)

警告:此功能已被弃用。它将在以后的版本中删除。更新说明:改为使用lite.TFLiteConverter

通常,此函数用于将TensorFlow GraphDef转换为TFLite。可以通过提供转发到build_toco_convert_protos的参数来自定义转换。此功能已被弃用。请改用lite.TFLiteConverter。

参数:

  • input_data:输入数据(经常是sess.graph_def),
  • input_tensors:输入张量列表。使用foo.get_shape()和foo.dtype计算类型和shape。
  • output_tensors:输出张量列表(仅使用.name)。
  • *args:参考build_toco_convert_protos,
  • **kwargs:参考build_toco_convert_protos。

返回:

转换后的数据。例如,如果TFLite是目的地,那么这将是字节数组中的tflite flatbuffer。

可能引发的异常:

定义在build_toco_convert_protos。

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号