下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

SetInferShapeRange

函数功能

注册ShapeRange推导函数。

函数原型

OpDef &SetInferShapeRange(gert::OpImplRegisterV2::InferShapeRangeKernelFunc func);

参数说明

参数

输入/输出

说明

func

输入

ShapeRange推导函数。InferShapeRangeKernelFunc类型定义如下:

using InferShapeRangeKernelFunc = UINT32 (*)(InferShapeRangeContext *);

返回值说明

OpDef算子定义,OpDef请参考OpDef类

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词