函数原型
def npu_onnx_graph_op(inputs, tout, model_path, name=None)
功能说明
以算子形式加载ONNX模型，最终转换成为GE Graph下沉到Device执行。
参数说明
参数名 |
输入/输出 |
描述 |
|---|---|---|
inputs |
输入 |
算子输入列表。 |
tout |
输入 |
输出类型列表。 |
model_path |
输入 |
ONNX模型所在路径及文件名，例如/test/test.onnx。 |
name |
输入 |
指定该算子的名称。 |
返回值
算子输出列表。
调用示例
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")