Home
last modified time | relevance | path

Searched defs:_conv_forward (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dconv.py51 …def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]) -> Tensor: # type:… member in _ConvNd
299 def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]): member in Conv1d
450 def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]): member in Conv2d
592 def _conv_forward(self, input: Tensor, weight: Tensor, bias: Optional[Tensor]): member in Conv3d
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dconv.cpp95 Tensor Conv2dImpl::_conv_forward(const Tensor& input, const Tensor& weight) { in _conv_forward() function in torch::nn::Conv2dImpl
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd.py157 def _conv_forward(self, x): member in AotAutogradFallbackTests.test_call_fn_with_non_const_inputs_aot_safe.ModuleSpecialFwd
H A Dtest_modules.py1624 def _conv_forward(self, x): member in NNModuleTests.test_call_fn_with_non_const_inputs_safe.ModuleSpecialFwd