注册DataType推导函数。
OpDef &SetInferDataType(gert::OpImplKernelRegistry::InferDataTypeKernelFunc func);
参数 |
输入/输出 |
说明 |
|---|---|---|
func |
输入 |
DataType推导函数。InferDataTypeKernelFunc类型定义如下： using InferDataTypeKernelFunc = UINT32 (*)(InferDataTypeContext *); |
OpDef算子定义，OpDef请参考OpDef类。
无