Home
last modified time | relevance | path

Searched defs:grad_output_t (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DConvShared.cpp310 const Tensor& grad_output_t, in cudnn_convolution_transpose_backward_input()
443 const Tensor& grad_output_t, in cudnn_convolution_backward_input()
505 const Tensor& grad_output_t, in cudnn_convolution_backward_weight()
558 const Tensor& grad_output_t, in cudnn_convolution_backward_weight()
583 const at::Tensor& grad_output_t, in cudnn_convolution_backward()
637 const Tensor& grad_output_t, in cudnn_convolution_transpose_backward_weight()
662 const at::Tensor& grad_output_t, in cudnn_convolution_transpose_backward()
H A DGridSampler.cpp30 const Tensor& grad_output_t) { in cudnn_grid_sampler_backward()
130 const Tensor& grad_output_t) { in cudnn_grid_sampler_backward()
H A DBatchNorm.cpp273 const Tensor& grad_output_t, in cudnn_batch_norm_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DConv_miopen.cpp926 const Tensor& grad_output_t) in miopen_convolution_backward_bias()
1101 const Tensor& grad_output_t, in miopen_depthwise_convolution_backward_weight()
1154 const Tensor& grad_output_t, in miopen_convolution_backward_weight()
1168 const Tensor& grad_output_t, const Tensor& weight_t, in miopen_convolution_transpose_backward_input()
1181 const Tensor& grad_output_t, in miopen_convolution_transpose_backward_weight()
1195 const at::Tensor& input, const at::Tensor& grad_output_t, const at::Tensor& weight, in miopen_convolution_transpose_backward()
1317 IntArrayRef input_size, const Tensor& grad_output_t, const Tensor& weight_t, in miopen_convolution_backward_input()
1411 IntArrayRef input_size, const Tensor& grad_output_t, const Tensor& weight_t, in miopen_depthwise_convolution_backward_input()
1424 const at::Tensor& input, const at::Tensor& grad_output_t, const at::Tensor& weight, in miopen_convolution_backward()
1445 const at::Tensor& input, const at::Tensor& grad_output_t, const at::Tensor& weight, in miopen_depthwise_convolution_backward()
H A DBatchNorm_miopen.cpp161 const Tensor& grad_output_t, in miopen_batch_norm_backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp416 auto grad_output_t = grad_output_a[t]; in slow_conv2d_backward_out_cpu_template() local
527 auto grad_output_t = grad_output_a[t]; in slow_conv2d_backward_weight_out_cpu_template() local
H A DConvolutionMM3d.cpp442 auto grad_output_t = grad_output_a[t]; in slow_conv3d_backward_out_cpu_template() local
544 auto grad_output_t = grad_output_a[t]; in slow_conv3d_backward_parameters_out_cpu_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DConv.cpp866 const Tensor& input_t, const Tensor& grad_output_t, const Tensor& weight_t, in mkldnn_convolution_backward()
1017 const Tensor& input_t, const Tensor& grad_output_t, const Tensor& weight_t, in mkldnn_convolution_transpose_backward()
H A DLinear.cpp43 const Tensor& input, const Tensor& grad_output_t, in mkldnn_linear_backward()