Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add device-agnostic NestedTensor Ops (#1258)
Device-agnostic ops. CUDA, CPU dispatched to the same ops as well. - `_nested_compute_contiguous_strides_offsets` - `_nested_tensor_from_mask` - `_nested_tensor_from_mask_left_aligned` - `_nested_view_from_buffer`
- Loading branch information