智能客服
你问我答,随时在线为你解决问题
封装算子的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。
- IMPLEMT_INFERFUNC(op_name, func_name)
无
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| op_name | 输入 | 算子类型。 |
| func_name | 输入 | InferShape函数名,开发者自定义。 |
无
智能客服
你问我答,随时在线为你解决问题
合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
解决方案
精准高效的一站式服务支持,助力开发者商业成功