获取tensor类型的属性值。
const Tensor *GetTensor(size_t index) const
参数 |
输入/输出 |
说明 |
---|---|---|
index |
输入 |
属性在IR原型定义中以及在OP_IMPL注册中的索引。 |
指向属性值的指针。
无。
const RuntimeAttrs * runtime_attrs = kernel_context->GetAttrs(); const Tensor *attr0 = runtime_attrs->GetTensor(0);