Searched defs:parametrization (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/ao/pruning/_experimental/pruner/ | ||
| D | parametrization.py | 7 def module_contains_param(module, parametrization): argument |
| /external/pytorch/torch/nn/utils/ | ||
| D | parametrize.py | 378 def get_cached_parametrization(parametrization) -> Tensor: argument |
| /external/pytorch/test/dynamo/ | ||
| D | test_activation_checkpointing.py | 1027 def parametrization(self, x): member in ActivationCheckpointingViaTagsTests.test_compile_selective_checkpoint_parametrization.Parametrization |
| /external/pytorch/test/nn/ | ||
| D | test_parametrization.py | 1759 def _check_parametrization( argument |