Sparse Operators ================ .. automodule:: fbgemm_gpu .. _sparse-ops-stable-api: Stable API ---------- .. autofunction:: torch.ops.fbgemm.permute_2D_sparse_data .. autofunction:: torch.ops.fbgemm.permute_1D_sparse_data .. autofunction:: torch.ops.fbgemm.expand_into_jagged_permute .. autofunction:: torch.ops.fbgemm.asynchronous_complete_cumsum .. autofunction:: torch.ops.fbgemm.offsets_range .. autofunction:: torch.ops.fbgemm.segment_sum_csr .. autofunction:: torch.ops.fbgemm.keyed_jagged_index_select_dim1 .. autofunction:: torch.ops.fbgemm.block_bucketize_sparse_features Other API