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