Home
last modified time | relevance | path

Searched defs:output_padding_depth (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNaiveConvolutionTranspose3d.cpp45 int output_padding_depth, in slow_conv_transpose3d_shape_check()
224 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_out_cpu_template() local
438 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_backward_out_cpu_template() local
643 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_acc_grad_parameters_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNaiveConvolutionTranspose3d.cu45 int output_padding_depth, in slow_conv_transpose3d_shape_check()
220 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_out_cuda_template() local
444 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_backward_out_cuda_template() local
652 int64_t output_padding_depth = output_padding[0]; in slow_conv_transpose3d_acc_grad_parameters_cuda() local