mindspore.ops.prim_attr_register
- mindspore.ops.prim_attr_register(fn)[source]
Primitive attributes register.
Register the decorator of the built-in operator primitive ‘__init__’. The function will add all the parameters of ‘__init__’ as operator attributes.
- Parameters
fn (function) – __init__ function of primitive.
- Returns
function, original function.