文档
注册
评分
提单
论坛
小AI

InferShapeFuncRegister构造函数和析构函数

函数功能

InferShapeFuncRegister构造函数和析构函数。

函数原型

数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。

InferShapeFuncRegister (const std::string &operator_type, const InferShapeFunc &infer_shape_func);

InferShapeFuncRegister(const char *const operator_type, const InferShapeFunc &infer_shape_func);

~ InferShapeFuncRegister()

参数说明

参数名

输入/输出

描述

operator_type

输入

算子类型。

infer_shape_func

输入

算子infershape函数。

返回值

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

约束说明

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

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

当前产品无相关内容

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