/aosp_15_r20/external/pytorch/torch/nn/modules/ |
H A D | lazy.py | 240 def initialize_parameters(self: _LazyProtocol, *args, **kwargs): member in LazyModuleMixin 274 module.initialize_parameters(*args, **kwargs)
|
H A D | linear.py | 283 def initialize_parameters(self, input) -> None: # type: ignore[override] member in LazyLinear
|
H A D | batchnorm.py | 250 def initialize_parameters(self, input) -> None: # type: ignore[override] member in _LazyNormBase
|
H A D | conv.py | 1156 … def initialize_parameters(self, input: Tensor, *args, **kwargs) -> None: # type: ignore[override] member in _LazyConvXdMixin
|
/aosp_15_r20/external/pytorch/test/dynamo/ |
H A D | test_modules.py | 580 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 D | test_lazy_modules.py | 249 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 D | test_subclass.py | 214 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 D | test_torch_export_with_onnxruntime.py | 291 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 D | test_export.py | 4999 def initialize_parameters(self, *args, **kwargs): member in TestExport.test_lazy_module_kwargs.LazyModule
|