文档首页我要评分文档获取效率文档正确性内容完整性文档易理解0/200提交在线提单论坛求助昇腾小AI SetInferShape 函数功能 注册Shape推导函数。 函数原型 OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func); 参数说明 参数 输入/输出 说明 func 输入 Shape推导函数。InferShapeKernelFunc类型定义如下: using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *); 返回值说明 OpDef算子定义,OpDef请参考OpDef。 约束说明 无 父主题: OpDef 搜索结果找到“0”个结果当前产品无相关内容未找到相关内容,请尝试其他搜索词