使用图模式时，需要调用该接口注册DataType推导函数。
1
|
OpDef &SetInferDataType(gert::OpImplRegisterV2::InferDataTypeKernelFunc func) |
|
参数 |
输入/输出 |
说明 |
||
|---|---|---|---|---|
|
func |
输入 |
DataType推导函数。InferDataTypeKernelFunc类型定义如下，入参类型参考InferDataTypeContext：
|
OpDef算子定义，OpDef请参考OpDef。
无