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.