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

InferShapeFuncRegister构造函数和析构函数

函数功能

InferShapeFuncRegister构造函数和析构函数。

函数原型

InferShapeFuncRegister (const std::string &operatorType, InferShapeFunc inferShapeFunc);

~ InferShapeFuncRegister()

参数说明

参数名

输入/输出

描述

operatorType

输入

算子类型。

inferShapeFunc

输入

算子infershape函数。

返回值

InferShapeFuncRegister构造函数返回InferShapeFuncRegister类型的对象。

约束说明

算子infershape函数注册接口,此接口被其他头文件引用,一般不用由算子开发者直接调用。

分享:

    相关文档

    相关产品

close