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

INFER_FORMAT_FUNC_REG

函数功能

注册算子的InferFormat实现。

GE会在整图的Shape与Dtype推导前后分别调用一次整图的InferFormat,过程中会分别调用各个算子的InferFormat函数。如果算子没有注册InferFormat函数,GE将使用默认的推导函数,即输出的Format等于输入的Format。

函数原型

#define INFER_FORMAT_FUNC_REG(op_name, x) \

__INFER_FORMAT_FUNC_REG_IMPL__(op_name, INFER_FORMAT_FUNC(op_name, x), __COUNTER__)

参数说明

参数名

输入/输出

描述

op_name

输入

算子类型。

x

输入

inferFormat函数名,使用IMPLEMT_INFERFORMAT_FUNC中的func_name

返回值

约束和限制说明

客户是否可用

调用示例和相关API

INFER_FORMAT_FUNC_REG(Transpose, TransposeInferFormat);
搜索结果
找到“0”个结果

当前产品无相关内容

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