Home
last modified time | relevance | path

Searched defs:grad_input_arg (Results 1 – 14 of 14) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DMaxUnpooling.cu411 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
DUpSampleNearest1d.cu153 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest1d_backward_out_cuda_template() local
DUpSampleLinear1d.cu169 grad_input_arg{grad_input, "grad_input", 2}; in upsample_linear1d_backward_out_cuda_template() local
DUpSampleBicubic2d.cu230 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_bicubic2d_backward_out_cuda_template() local
DUpSampleNearest3d.cu230 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest3d_backward_out_cuda_template() local
DUpSampleTrilinear3d.cu307 TensorArg grad_input_arg{grad_input_, "grad_input_", 1}, in upsample_trilinear3d_backward_out_cuda_template() local
DAdaptiveMaxPooling2d.cu331 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in TORCH_IMPL_FUNC() local
DUpSampleBilinear2d.cu367 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
DUpSampleNearest2d.cu342 TensorArg grad_input_arg{grad_input, "grad_input", 1}, in upsample_nearest2d_backward_out_cuda_template() local
DNaiveConvolutionTranspose2d.cu341 grad_input_arg{grad_input, "grad_input", 4}; in slow_conv_transpose2d_backward_out_cuda_template() local
DAdaptiveMaxPooling3d.cu390 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in TORCH_IMPL_FUNC() local
DAdaptiveAveragePooling3d.cu426 TensorArg grad_input_arg{gradInput, "gradInput", 1}; in adaptive_avg_pool3d_backward_out_cuda_template() local
DNaiveConvolutionTranspose3d.cu451 grad_input_arg{grad_input, "grad_input", 4}; in slow_conv_transpose3d_backward_out_cuda_template() local
DAdaptiveAveragePooling.cu606 TensorArg grad_input_arg{ gradInput, "gradInput", 1 }, in adaptive_avg_pool2d_backward_out_cuda_template() local