Home
last modified time | relevance | path

Searched refs:initialize_parameters (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dlazy.py240 def initialize_parameters(self: _LazyProtocol, *args, **kwargs): member in LazyModuleMixin
274 module.initialize_parameters(*args, **kwargs)
H A Dlinear.py283 def initialize_parameters(self, input) -> None: # type: ignore[override] member in LazyLinear
H A Dbatchnorm.py250 def initialize_parameters(self, input) -> None: # type: ignore[override] member in _LazyNormBase
H A Dconv.py1156 … def initialize_parameters(self, input: Tensor, *args, **kwargs) -> None: # type: ignore[override] member in _LazyConvXdMixin
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py580 def initialize_parameters(self, x): member in LazyModule
609 def initialize_parameters(self, input): member in LazyLayerWithNamedTupleInput
636 def initialize_parameters(self, input): member in LazyLayerWithListInput
660 def initialize_parameters(self, input): member in LazyModuleWithLazySubmodule
672 def initialize_parameters(self, x, y): member in LazyLayerWithInputs
691 def initialize_parameters(self, *args, **kwargs): member in LazyModuleKwArgs
714 def initialize_parameters(self, input): member in LazyChildModuleNoClsToBecome
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_lazy_modules.py249 def initialize_parameters(self, input): member in TestLazyModules.test_lazy_pre_forward_hook.TestModule
270 def initialize_parameters(self, input): member in TestLazyModules.test_lazy_forward_hook.TestModule
/aosp_15_r20/external/pytorch/test/
H A Dtest_subclass.py214 def initialize_parameters(self, input) -> None: # type: ignore[override] member in TestSubclass.test_lazy_module.MyLazyModule
/aosp_15_r20/external/pytorch/test/onnx/torch_export/
H A Dtest_torch_export_with_onnxruntime.py291 def initialize_parameters(self, *args, **kwargs): member in TestFxToOnnxWithOnnxRuntime.test_onnx_program_supports_lazy_module_kwargs.LazyModule
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_export.py4999 def initialize_parameters(self, *args, **kwargs): member in TestExport.test_lazy_module_kwargs.LazyModule