Searched refs:complex_convolution_mode (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Convolution.cpp | 883 at::Tensor complex_convolution_mode( in complex_convolution_mode() function
1081 output = complex_convolution_mode(input, weight, bias, stride, padding, dilation, groups); in conv1d_padding_symint()
1095 output = complex_convolution_mode(input, weight, bias, stride, padding, dilation, groups); in conv2d_padding_symint()
1109 output = complex_convolution_mode(input, weight, bias, stride, padding, dilation, groups); in conv3d_padding_symint()
|