Atlas 500应用-INFER_SHAPE_AND_TYPE:DECLARE_INFERFUNC和IMPLEMT_INFERFUNC宏说明

时间:2023-11-01 16:13:36

DECLARE_INFERFUNC和IMPLEMT_INFERFUNC宏说明

在注册用于推理算子的Shape和DataType的函数前,需要先用DECLARE_INFERFUNC宏申明函数、用IMPLEMT_INFERFUNC宏定义函数。

  • 声明函数
    DECLARE_INFERFUNC(FullConnection, FullConnectionInfer)

    DECLARE_INFERFUNC宏展开后的实现为:

    namespace op {class FullConnection;}static graphStatus FullConnectionInfer(op::FullConnection& op);
  • 定义函数
    IMPLEMT_INFERFUNC(FullConnection, FullConnectionInfer) {// 实现细节}

    IMPLEMT_INFERFUNC宏展开后的实现为:

    static graphStatus FullConnectionInfer(op::FullConnection& op){// 实现细节}
support.huaweicloud.com/api-ge-atlas500app/atlasge_07_0091.html