Home
last modified time | relevance | path

Searched refs:_rebuild_nested_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/
H A D_weights_only_unpickler.py118 torch._utils._rebuild_nested_tensor,
H A D_utils.py329 def _rebuild_nested_tensor(buffer, sizes, strides, storage_offsets): function
H A D_tensor.py414 return (torch._utils._rebuild_nested_tensor, args_nested)