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

COMMON_INFER_FUNC_REG

函数功能

注册算子的InferShape函数。

INFER_FUNC_REG的区别是,此函数注册的InferShape函数入参为operator基类而非子类,此接口支持多算子共用同一个InferShape函数。

函数原型

COMMON_INFER_FUNC_REG(op_name, x)

约束说明

无。

参数说明

参数名

输入/输出

描述

op_name

输入

算子类型。

x

输入

InferShape函数名,和IMPLEMT_COMMON_INFERFUNC的InferShape函数名保持一致。

返回值

无。

分享:

    相关文档

    相关产品

关闭导读