GetOptionalInputShapeRange

函数功能

基于算子IR原型定义,获取`OPTIONAL_INPUT`类型的输入shape range指针。

函数原型

const Range<Shape> *GetOptionalInputShapeRange(const size_t ir_index) const;

参数说明

参数

输入/输出

说明

ir_index

输入

算子IR原型定义中的输入索引,从0开始计数。

返回值说明

返回shape range指针,index非法,或该INPUT没有实例化时,返回空指针。

约束说明

调用示例

ge::graphStatus InferShapeForXXX(InferShapeContext *context) {
  auto in_shape = context->GetOptionalInputShapeRange(2);
  // ...
}