GetAttr
函数功能
获取算子的属性。
输入参数name需要为有效值。若获取失败,则返回空指针。
函数原型
AttrValue *GetAttr(std::string name) const
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
name |
输入 |
支持以下输入:
|
返回值说明
AttrValue *,返回的AttrValue的指针。若获取失败,则返回空指针。
约束说明
无。
调用示例
- 示例:获取属性“seed”的值
AttrValue *seed0 = ctx.GetAttr("seed");
- 并行计算场景下,获取分块的数量及本次计算的分块ID
uint32_t blockdim = ctx.GetAttr("block_num")->GetInt(); uint32_t blockid = ctx.GetAttr("block_id")->GetInt();
父主题: CpuKernelContext类