InferDataType

函数功能

注册算子的InferDataType函数。

用户需要为算子编写一个InferDataTypeKernelFunc类型的函数，并使用该接口进行注册。

InferDataTypeKernelFunc类型定义如下：

using InferDataTypeKernelFunc = UINT32 (*)(InferDataTypeContext *);

函数原型

OpImplRegisterV2 &InferDataType(InferDataTypeKernelFunc infer_datatype_func);

参数说明

返回值说明

返回算子的OpImplRegisterV2对象，该对象新增注册了InferDataType函数infer_datatype_func。

约束说明

无。