Searched defs:grad_output_t (Results 1 – 9 of 9) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | ConvShared.cpp | 310 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()
|
| D | GridSampler.cpp | 30 const Tensor& grad_output_t) { in cudnn_grid_sampler_backward()
130 const Tensor& grad_output_t) { in cudnn_grid_sampler_backward()
|
| D | BatchNorm.cpp | 273 const Tensor& grad_output_t, in cudnn_batch_norm_backward()
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | Conv_miopen.cpp | 926 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()
|
| D | BatchNorm_miopen.cpp | 161 const Tensor& grad_output_t, in miopen_batch_norm_backward()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | ConvolutionMM2d.cpp | 416 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
|
| D | ConvolutionMM3d.cpp | 442 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
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Linear.cpp | 43 const Tensor& input, const Tensor& grad_output_t, in mkldnn_linear_backward()
|
| D | Conv.cpp | 866 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()
|