npu_onnx_graph_op

产品支持情况

产品	是否支持
Atlas A2 训练系列产品	√
Atlas 800I A2 推理产品/A200I A2 Box 异构组件	x
Atlas 200I/500 A2 推理产品	x
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	x

针对 Atlas 推理系列产品，仅支持TensorFlow 1.15网络的在线推理特性。

功能说明

以算子形式加载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。

返回值

算子输出列表。

调用示例

      
           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")

父主题： npu_bridge.estimator.npu_ops