mindspore.ops.get_vm_impl_fn

mindspore.ops.get_vm_impl_fn(prim)

Gets the virtual implementation function by a primitive object or primitive name.

Parameters

prim (Union[Primitive, str]) – primitive object or name for operator register.

Returns

function, vm function