Home
last modified time | relevance | path

Searched defs:grad_input_t (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DGridSampler.cpp149 auto grad_input_t = at::empty({0}, input->options()); in cudnn_grid_sampler_backward() local
H A DBatchNorm.cpp327 auto grad_input_t = at::empty( in cudnn_batch_norm_backward() local
H A DConvShared.cpp381 Tensor grad_input_t = at::detail::empty_cuda( in cudnn_convolution_backward_input() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp211 auto grad_input_t = at::empty(input->sizes(), input->options()); in miopen_batch_norm_backward() local
H A DConv_miopen.cpp1282 Tensor grad_input_t = at::detail::empty_cuda( in miopen_convolution_backward_input() local
1389 Tensor grad_input_t = at::detail::empty_cuda( in miopen_depthwise_convolution_backward_input() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DConvolutionMM2d.cpp415 auto grad_input_t = grad_input_a[t]; in slow_conv2d_backward_out_cpu_template() local
H A DConvolutionMM3d.cpp441 auto grad_input_t = grad_input_a[t]; in slow_conv3d_backward_out_cpu_template() local