| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MaxUnpooling.cu | 411 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in max_unpooling2d_backward_out_cuda() local
523 grad_input_arg{grad_input, "grad_input", 4}; in max_unpooling3d_backward_out_cuda() local
|
| H A D | UpSampleBilinear2d.cu | 367 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_bilinear2d_backward_out_cuda_template() local
779 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_gen2d_aa_backward_out_cuda_template() local
|
| H A D | UpSampleNearest1d.cu | 153 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest1d_backward_out_cuda_template() local
|
| H A D | UpSampleLinear1d.cu | 169 grad_input_arg{grad_input, "grad_input", 2}; in upsample_linear1d_backward_out_cuda_template() local
|
| H A D | UpSampleBicubic2d.cu | 230 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_bicubic2d_backward_out_cuda_template() local
|
| H A D | UpSampleNearest3d.cu | 230 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest3d_backward_out_cuda_template() local
|
| H A D | AdaptiveMaxPooling2d.cu | 331 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in TORCH_IMPL_FUNC() local
|
| H A D | UpSampleTrilinear3d.cu | 307 TensorArg grad_input_arg{grad_input_, "grad_input_", 1}, in upsample_trilinear3d_backward_out_cuda_template() local
|
| H A D | UpSampleNearest2d.cu | 342 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest2d_backward_out_cuda_template() local
|
| H A D | NaiveConvolutionTranspose2d.cu | 341 grad_input_arg{grad_input, "grad_input", 4}; in slow_conv_transpose2d_backward_out_cuda_template() local
|
| H A D | AdaptiveMaxPooling3d.cu | 390 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in TORCH_IMPL_FUNC() local
|
| H A D | AdaptiveAveragePooling3d.cu | 426 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in adaptive_avg_pool3d_backward_out_cuda_template() local
|
| H A D | NaiveConvolutionTranspose3d.cu | 451 grad_input_arg{grad_input, "grad_input", 4}; in slow_conv_transpose3d_backward_out_cuda_template() local
|
| H A D | AdaptiveAveragePooling.cu | 606 TensorArg grad_input_arg{ gradInput, "gradInput", 1 }, in adaptive_avg_pool2d_backward_out_cuda_template() local
|