InferShapeFuncRegister构造函数和析构函数
函数功能
InferShapeFuncRegister构造函数和析构函数。
函数原型
InferShapeFuncRegister (const std::string &operatorType, InferShapeFunc inferShapeFunc);
~ InferShapeFuncRegister()
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
operatorType |
输入 |
算子类型。 |
inferShapeFunc |
输入 |
算子infershape函数。 |
返回值
InferShapeFuncRegister构造函数返回InferShapeFuncRegister类型的对象。
约束说明
算子infershape函数注册接口,此接口被其他头文件引用,一般不用由算子开发者直接调用。
