下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

npu_onnx_graph_op

功能说明

以算子形式加载ONNX模型,将指定路径中onnx模型通过TensorFlow的框架执行在昇腾AI处理器

函数原型

def npu_onnx_graph_op(inputs, tout, model_path, name=None)

参数说明

参数名

输入/输出

描述

inputs

输入

onnx模型的输入,类型为tensor。

tout

输入

onnx模型的输出类型,如tf.float32。

model_path

输入

ONNX模型所在路径及文件名,例如/test/test.onnx,类型string。

name

输入

指定该算子在图上的名称,类型string。

返回值

算子输出列表。

调用示例

1
2
3
4
from npu_bridge.estimator.npu_ops import npu_onnx_graph_op

input = tf.placeholder(dtype=tf.float32, shape=(1, 1, 5, 5), name="conv_input")
output = npu_onnx_graph_op([input], [tf.float32], model_path="conv2d.onnx", name="conv2d")
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词