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

























根据算子原型定义中的输入索引获取对应的可选输入tensor指针。
- const Tensor *GetOptionalInputTensor(const size_t ir_index) const;
| 参数 | 输入/输出 | 说明 |
|---|---|---|
| ir_index | 输入 | 算子IR原型定义中的输入索引,从0开始计数。 |
如果输入没有被设置为数据依赖,调用此接口获取tensor时,只能在tensor中获取到正确的shape、format、datatype信息,无法获取到真实的tensor数据地址(获取到的地址为nullptr)。
- ge::graphStatus Tiling4ReduceCommon(TilingContext* context) {
- auto in_shape = context->GetInputShape(0);
- GE_ASSERT_NOTNULL(in_shape);
- auto axes_tensor = context->GetOptionalInputTensor(1);
- // ...
- }
智能客服
你问我答,随时在线为你解决问题
合作咨询
我们的专家服务团队将竭诚为您提供专业的合作咨询服务
解决方案
精准高效的一站式服务支持,助力开发者商业成功