更新时间:2021-03-18 GMT+08:00
分享

INFER_SHAPE_AND_TYPE

函数原型

INFER_SHAPE_AND_TYPE (x)

功能说明

注册用于推理算子的Shape和DataType的函数。

参数说明

参数名

输入/输出

类型

描述

x

输入

-

宏参数,算子Shape和DataType推理函数。

例如,INFER_SHAPE_AND_TYPE(FullConnectionInfer)用于注册FullConnectionInfer函数,用于推理算子的Shape和DataType。

FullConnectionInfer通过DECLARE_INFERFUNC声明,通过IMPLEMT_INFERFUNC定义,详情请参见DECLARE_INFERFUNC和IMPLEMT_INFERFUNC宏说明

返回值

无。

异常处理

无。

约束说明

无。

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){
    // 实现细节
    }
分享:

    相关文档

    相关产品