Home
last modified time | relevance | path

Searched defs:leave_parametrized (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/
H A Dbase_data_sparsifier.py287 def squash_mask(self, *args, leave_parametrized=True, names=None, **kwargs): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_subclass.py173 def test_parametrization(self, tensor_cls, leave_parametrized): argument
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_parametrization.py1763 leave_parametrized=False, argument