基于算子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); // ... }