开发者
下载
[object Object][object Object]

全量芯片支持。

[object Object]

#include <graph/operator_reg.h>

[object Object]

封装算子的InferShape函数。

该函数传入的OpType为基于Operator类派生出来的子类,会自动生成一个类型为此子类的对象op,可以使用子类的成员函数获取输入输出描述的方法,从而进行InferShape的实现。

基于OpType派生出来的子类op的成员函数如下:

  • op.set_input_x(Operator &v, const string &srcName):将网络中算子v的输出srcName设置为当前算子的输入x。

  • op.get_input_desc_x():获取该算子的输入x的描述信息,返回对象为TensorDesc类型。

    op.update_input_desc_x(const TensorDesc& tensorDesc):更新输入x的描述信息,包括shape、datatype与format。

  • op.get_output_desc_y():获取该算子的输出y的描述信息,返回对象TensorDesc类型。

  • op.update_output_desc_y(const TensorDesc& tensorDesc):更新输出y的描述信息,包括shape、datatype与format。

  • op.get_attr_attr1(AscendString &val):获取算子属性attr1的值val。

[object Object]
[object Object]
[object Object][object Object]undefined
[object Object]

[object Object]