(
rebuild_buffer_func,
rebuild_buffer_args,
rebuild_sizes_func,
rebuild_sizes_args,
rebuild_strides_func,
rebuild_strides_args,
rebuild_offsets_func,
rebuild_offsets_args,
)
| 404 | |
| 405 | |
| 406 | def rebuild_nested_tensor( |
| 407 | rebuild_buffer_func, |
| 408 | rebuild_buffer_args, |
| 409 | rebuild_sizes_func, |
| 410 | rebuild_sizes_args, |
| 411 | rebuild_strides_func, |
| 412 | rebuild_strides_args, |
| 413 | rebuild_offsets_func, |
| 414 | rebuild_offsets_args, |
| 415 | ): |
| 416 | buffer = rebuild_buffer_func(*rebuild_buffer_args) |
| 417 | sizes = rebuild_sizes_func(*rebuild_sizes_args) |
| 418 | strides = rebuild_strides_func(*rebuild_strides_args) |
| 419 | offsets = rebuild_offsets_func(*rebuild_offsets_args) |
| 420 | return torch._nested_view_from_buffer_copy(buffer, sizes, strides, offsets) |
| 421 | |
| 422 | |
| 423 | def reduce_nested_tensor(nt): |
nothing calls this directly
no outgoing calls
no test coverage detected