Searched refs:OpsWrapper (Results 1 – 2 of 2) sorted by relevance
256 class OpsWrapper: class263 new_args = [OpsWrapper._unwrap(a) for a in args]264 new_kwargs = {k: OpsWrapper._unwrap(v) for k, v in kwargs.items()}265 return OpsWrapper._wrap(getattr(_ops, name)(*new_args, **new_kwargs))272 return tuple(OpsWrapper._unwrap(v) for v in x)286 index = OpsWrapper._unwrap(index)290 ops = OpsWrapper()
52 from ..virtualized import ops, OpsWrapper, V750 mask = OpsWrapper._unwrap(mask)927 return OpsWrapper._unwrap((mean, m2, rnumel))