Searched refs:wrap_numpy (Results 1 – 4 of 4) sorted by relevance
257 def wrap_numpy(fn): function284 from torch._dynamo.external_utils import wrap_numpy as wrap
57 def wrap_numpy(f): function
517 We can do so by using ``torch.compiler.wrap_numpy``:522 @torch.compiler.wrap_numpy534 ``wrap_numpy`` is in charge of marking any ``torch.Tensor`` input as an input544 @torch.compiler.wrap_numpy569 @torch.compiler.wrap_numpy
10912 return torch.compiler.wrap_numpy(my_np)(x)