Searched refs:convolution_shape_check (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | ConvShared.cpp | 221 convolution_shape_check( in cudnn_convolution_forward_out()
386 convolution_shape_check( in cudnn_convolution_backward_input()
531 convolution_shape_check( in cudnn_convolution_backward_weight()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | Conv_miopen.cpp | 779 convolution_shape_check(c, input, weight, output, padding, stride, dilation, groups); in miopen_convolution_forward()
882 convolution_shape_check(c, input, weight, output, padding, stride, dilation, groups); in miopen_depthwise_convolution_forward()
1090 …convolution_shape_check(c, input, grad_weight, grad_output_contig, padding, stride, dilation, grou… in miopen_depthwise_convolution_backward_weight()
1143 …convolution_shape_check(c, input, grad_weight, grad_output_contig, padding, stride, dilation, grou… in miopen_convolution_backward_weight()
1287 convolution_shape_check(c, grad_input, weight, grad_output, padding, stride, dilation, groups); in miopen_convolution_backward_input()
1393 convolution_shape_check(c, grad_input, weight, grad_output, padding, stride, dilation, groups); in miopen_depthwise_convolution_backward_input()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | Convolution.mm | 178 convolution_shape_check(c, input, weight, output, padding, stride, dilation, groups);
400 convolution_shape_check(c, grad_input, weight, grad_output, padding, stride, dilation, groups);
562 convolution_shape_check(c, input, grad_weight, grad_output, padding, stride, dilation, groups);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ConvUtils.h | 191 inline void convolution_shape_check( in convolution_shape_check() function
|