| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | transformer.cpp | 48 void TransformerEncoderLayerImpl::reset_parameters() { in reset_parameters() function in torch::nn::TransformerEncoderLayerImpl
146 void TransformerDecoderLayerImpl::reset_parameters() { in reset_parameters() function in torch::nn::TransformerDecoderLayerImpl
237 void TransformerEncoderImpl::reset_parameters() { in reset_parameters() function in torch::nn::TransformerEncoderImpl
304 void TransformerDecoderImpl::reset_parameters() { in reset_parameters() function in torch::nn::TransformerDecoderImpl
411 void TransformerImpl::reset_parameters() { in reset_parameters() function in torch::nn::TransformerImpl
|
| H A D | embedding.cpp | 51 void EmbeddingImpl::reset_parameters() { in reset_parameters() function in torch::nn::EmbeddingImpl
127 void EmbeddingBagImpl::reset_parameters() { in reset_parameters() function in torch::nn::EmbeddingBagImpl
|
| H A D | normalization.cpp | 34 void LayerNormImpl::reset_parameters() { in reset_parameters() function in torch::nn::LayerNormImpl
106 void GroupNormImpl::reset_parameters() { in reset_parameters() function in torch::nn::GroupNormImpl
|
| H A D | linear.cpp | 45 void LinearImpl::reset_parameters() { in reset_parameters() function in torch::nn::LinearImpl
154 void BilinearImpl::reset_parameters() { in reset_parameters() function in torch::nn::BilinearImpl
|
| H A D | rnn.cpp | 280 void RNNImplBase<Derived>::reset_parameters() { in reset_parameters() function in torch::nn::detail::RNNImplBase
838 void RNNCellImplBase<Derived>::reset_parameters() { in reset_parameters() function in torch::nn::detail::RNNCellImplBase
|
| H A D | adaptive.cpp | 68 void AdaptiveLogSoftmaxWithLossImpl::reset_parameters() { in reset_parameters() function in torch::nn::AdaptiveLogSoftmaxWithLossImpl
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | linear.py | 114 def reset_parameters(self) -> None: member in Linear
218 def reset_parameters(self) -> None: member in Bilinear
279 def reset_parameters(self) -> None: member in LazyLinear
|
| H A D | normalization.py | 210 def reset_parameters(self) -> None: member in LayerNorm
307 def reset_parameters(self) -> None: member in GroupNorm
390 def reset_parameters(self) -> None: member in RMSNorm
|
| H A D | sparse.py | 180 def reset_parameters(self) -> None: member in Embedding
418 def reset_parameters(self) -> None: member in EmbeddingBag
|
| H A D | batchnorm.py | 93 def reset_parameters(self) -> None: member in _NormBase
246 def reset_parameters(self) -> None: member in _LazyNormBase
|
| H A D | adaptive.py | 173 def reset_parameters(self) -> None: member in AdaptiveLogSoftmaxWithLoss
|
| H A D | rnn.py | 292 def reset_parameters(self) -> None: member in RNNBase
1484 def reset_parameters(self) -> None: member in RNNCellBase
|
| H A D | conv.py | 141 def reset_parameters(self) -> None: member in _ConvNd
1147 def reset_parameters(self) -> None: member in _LazyConvXdMixin
|
| H A D | activation.py | 1508 def reset_parameters(self): member in PReLU
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_meta.py | 66 def reset_parameters(self, *args, **kwargs): member in MyLinear
78 def reset_parameters(self, *args, **kwargs): member in MyBuffer
|
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
| H A D | sample_module.py | 28 def reset_parameters(self): member in SampleModule
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | batchnorm.h | 64 void reset_parameters() { in reset_parameters() function
|
| H A D | conv.h | 104 void reset_parameters() { in reset_parameters() function
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_generator.py | 142 def reset_parameters(self): member in TestGenerator.test_save_load.Foo
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| H A D | linear_fused.py | 79 def reset_parameters(self): member in LinearBn1d
|
| /aosp_15_r20/external/pytorch/benchmarks/transformer/ |
| H A D | attention_bias_benchmarks.py | 136 def reset_parameters(self): member in CompositeMHA
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_api.py | 31 def reset_parameters(self): member in MyModel
|
| H A D | test_dtensor.py | 50 def reset_parameters(self, *args, **kwargs): member in DummyMLP
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 75 def reset_parameters(self): member in MLPModule
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 123 def reset_parameters(self): member in _ReferenceConvBnNd
|