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。