mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

[sparse] remove deprecated _dedupe() method

Browse Source
This commit is contained in:
Jake VanderPlas 2022-08-01 16:59:25 -07:00
parent 6b1610ce9e
commit 3cce03554f
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jax/experimental/sparse/bcoo.py
View File

@ -2139,10 +2139,6 @@ class BCOO(JAXSparse):
return cls((data, indices), shape=(N, M), indices_sorted=True,
unique_indices=True)
def _dedupe(self):
warnings.warn("_dedupe() is deprecated. Use sum_duplicates() instead.", FutureWarning)
return self.sum_duplicates(nse=self.nse)
def update_layout(self, *, n_batch=None, n_dense=None, on_inefficient='error'):
"""Update the storage layout (i.e. n_batch & n_dense) of a BCOO matrix.