Searched defs:parametrization (Results 1 – 6 of 6) sorted by relevance
7 def module_contains_param(module, parametrization): argument43 def __init__(self, parametrization, prune_bias): argument
378 def get_cached_parametrization(parametrization) -> Tensor: argument
1027 def parametrization(self, x): member in ActivationCheckpointingViaTagsTests.test_compile_selective_checkpoint_parametrization.Parametrization
1759 def _check_parametrization( argument
META-INF/ META-INF/MANIFEST.MF com/ com/github/ com/ ...