智能客服
你问我答,随时在线为你解决问题

























根据算子原型定义中的输入索引获取对应的可选输入shape指针。
- const StorageShape *GetOptionalInputShape(const size_t ir_index) const;
| 参数 | 输入/输出 | 说明 |
|---|---|---|
| ir_index | 输入 | 算子IR原型定义中的输入索引,从0开始计数。 |
指定的输入shape指针,shape中包含了原始shape与运行时shape。关于StorageShape类型的定义,请参见StorageShape。
当输入ir_index非法,或该INPUT没有实例化时,返回空指针。
无
- ge::graphStatus Tiling4ConcatD(TilingContext* context) {
- const StorageShape *shape_bias = context->GetOptionalInputShape(kBatchMatMulBiasIdx);
- // ...
- }
智能客服
你问我答,随时在线为你解决问题
合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
解决方案
精准高效的一站式服务支持,助力开发者商业成功