文档管理中心

函数功能

根据算子原型定义中的输入索引获取对应的可选输入shape指针。

函数原型

收起
自动换行
深色代码主题
复制
  1. const StorageShape *GetOptionalInputShape(const size_t ir_index) const;

参数说明

展开
参数 输入/输出 说明
ir_index 输入 算子IR原型定义中的输入索引,从0开始计数。

返回值

指定的输入shape指针,shape中包含了原始shape与运行时shape。关于StorageShape类型的定义,请参见StorageShape

当输入ir_index非法,或该INPUT没有实例化时,返回空指针。

约束说明

调用示例

收起
自动换行
深色代码主题
复制
  1. ge::graphStatus Tiling4ConcatD(TilingContext* context) {
  2. const StorageShape *shape_bias = context->GetOptionalInputShape(kBatchMatMulBiasIdx);
  3. // ...
  4. }
在 指南 中进行搜索
请输入您想要搜索的关键词