Home
last modified time | relevance | path

Searched defs:grad_output_a (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv2d.cu503 auto grad_output_a = grad_output.packed_accessor32<const scalar_t, 4>(); in conv_depthwise2d_backward_out() local
634 const auto grad_output_a = grad_output.packed_accessor32<const scalar_t, 4>(); in conv_depthwise2d_grad_weight_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp408 auto grad_output_a = grad_output.accessor<const scalar_t, 4>(); in slow_conv2d_backward_out_cpu_template() local
522 auto grad_output_a = grad_output.accessor<const scalar_t, 4>(); in slow_conv2d_backward_weight_out_cpu_template() local
H A DConvolutionMM3d.cpp434 auto grad_output_a = grad_output_contiguous.accessor<const scalar_t, 5>(); in slow_conv3d_backward_out_cpu_template() local
541 auto grad_output_a = grad_output_contiguous.accessor<const scalar_t, 5>(); in slow_conv3d_backward_parameters_out_cpu_template() local