Searched defs:_check_input_dim (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | batchnorm.py | 99 def _check_input_dim(self, input): member in _NormBase
339 def _check_input_dim(self, input): member in BatchNorm1d
373 def _check_input_dim(self, input): member in LazyBatchNorm1d
450 def _check_input_dim(self, input): member in BatchNorm2d
484 def _check_input_dim(self, input): member in LazyBatchNorm2d
561 def _check_input_dim(self, input): member in BatchNorm3d
595 def _check_input_dim(self, input): member in LazyBatchNorm3d
720 def _check_input_dim(self, input): member in SyncBatchNorm
|
| H A D | instancenorm.py | 37 def _check_input_dim(self, input): member in _InstanceNorm
199 def _check_input_dim(self, input): member in InstanceNorm1d
236 def _check_input_dim(self, input): member in LazyInstanceNorm1d
315 def _check_input_dim(self, input): member in InstanceNorm2d
353 def _check_input_dim(self, input): member in LazyInstanceNorm2d
431 def _check_input_dim(self, input): member in InstanceNorm3d
469 def _check_input_dim(self, input): member in LazyInstanceNorm3d
|
| H A D | padding.py | 37 def _check_input_dim(self, input): member in _CircularPadNd
92 def _check_input_dim(self, input): member in CircularPad1d
151 def _check_input_dim(self, input): member in CircularPad2d
200 def _check_input_dim(self, input): member in CircularPad3d
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | batchnorm.cpp | 17 void BatchNorm1dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::BatchNorm1dImpl
25 void BatchNorm2dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::BatchNorm2dImpl
30 void BatchNorm3dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::BatchNorm3dImpl
|
| H A D | instancenorm.cpp | 7 void InstanceNorm1dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::InstanceNorm1dImpl
14 void InstanceNorm2dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::InstanceNorm2dImpl
21 void InstanceNorm3dImpl::_check_input_dim(const Tensor& input) { in _check_input_dim() function in torch::nn::InstanceNorm3dImpl
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/modules/ |
| H A D | batchnorm.py | 65 def _check_input_dim(self, input): member in BatchNorm2d
104 def _check_input_dim(self, input): member in BatchNorm3d
|