Skip to content

yangchengtest/onnx-tf-atlas

Repository files navigation

特点

支持TENSORFLOW NHWC的数据通道推理,规避华为算子问题。
基于开源: https://github.com/onnx/onnx-tensorflow

使用方式:
CUDA_VISIBLE_DEVICES=11 python //使用CPU进行TF运算推理。
import onnx
from onnx_tf.backend import prepare
onnx_model = onnx.load("../models/xxx.onnx")
tf_rep = prepare(onnx_model,strict=False,input_format="NHWC")

缺点:

1.当前不支持TRANSPOSE,中间层TRANSPOSE数据处理会乱。
2.无法规避华为不支持的算子,如ADDV2等,请自己手动修改。

About

onnx-tf support huawei atlas

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published