下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

INFER_FUNC_REG

函数功能

注册算子的InferShape函数。

函数原型

INFER_FUNC_REG(op_name, x)

该函数内部会自动调用INFER_VERIFY_FUNC(op_name, x),INFER_VERIFY_FUNC函数中的op_name为算子的类型,x为指向INFER_FUNC_REG(op_name,x)中“x”的指针。

约束说明

无。

参数说明

参数名

输入/输出

描述

op_name

输入

算子类型。

x

输入

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

返回值

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词