Searched refs:_rebuild_nested_tensor (Results 1 – 3 of 3) sorted by relevance
118 torch._utils._rebuild_nested_tensor,
329 def _rebuild_nested_tensor(buffer, sizes, strides, storage_offsets): function
414 return (torch._utils._rebuild_nested_tensor, args_nested)