Home
last modified time | relevance | path

Searched refs:jagged_torch_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dnested_tensor.py304 from .ops import jagged_torch_function
310 return jagged_torch_function(func, *args, **kwargs)
H A Dops.py311 def jagged_torch_function(func, *args, **kwargs): function