文档首页>
Atlas 200 DK 开发者套件(1.0.7.alpha)>
TBE自定义算子开发指南>
接口参考>
插件和原型定义接口>
算子原型接口>
其他独立宏>
GET_DYNAMIC_INPUT_SHAPE
更新时间:2021-03-18 GMT+08:00
GET_DYNAMIC_INPUT_SHAPE
函数功能
获取算子指定动态输入的shape,用于实现InferShape函数。
函数原型
GET_DYNAMIC_INPUT_SHAPE(op, name, index)
约束说明
无。
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
op |
输入 |
算子对象。 |
name |
输入 |
算子输入。 |
index |
输入 |
指定动态输入的第index个输入。 |
返回值
shape值。
调用示例
提供公共函数宏封装供算子开发InferShape函数。该函数获取动态输入的shape
父主题: 其他独立宏
