Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose2d.cu408 bool need_columns = (kernel_height != 1 || kernel_width != 1 || stride_height != 1 || in slow_conv_transpose2d_backward_out_cuda_template() local
598 bool need_columns = (kernel_height != 1 || kernel_width != 1 || stride_height != 1 || in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
H A DNaiveConvolutionTranspose3d.cu518 bool need_columns = (kernel_depth != 1 || kernel_height != 1 || kernel_width != 1 || in slow_conv_transpose3d_backward_out_cuda_template() local
739 bool need_columns = (kernel_depth != 1 || kernel_height != 1 || kernel_width != 1 || in slow_conv_transpose3d_acc_grad_parameters_cuda() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose2d.cpp486 bool need_columns = (kernel_height != 1 || kernel_width != 1 || stride_height != 1 || in slow_conv_transpose2d_backward_out_cpu_template() local
678 bool need_columns = (kernel_height != 1 || kernel_width != 1 || stride_height != 1 || in slow_conv_transpose2d_acc_grad_parameters_cpu() local
H A DNaiveConvolutionTranspose3d.cpp509 bool need_columns = (kernel_depth != 1 || kernel_height != 1 || kernel_width != 1 || in slow_conv_transpose3d_backward_out_cpu_template() local
722 bool need_columns = (kernel_depth != 1 || kernel_height != 1 || kernel_width != 1 || in slow_conv_transpose3d_acc_grad_parameters_cpu() local