Home
last modified time | relevance | path

Searched defs:grad_output_arg (Results 1 – 15 of 15) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DNaiveConvolutionTranspose2d.cu339 grad_output_arg{grad_output_, "grad_output", 2}, in slow_conv_transpose2d_backward_out_cuda_template() local
518 grad_output_arg{grad_output_, "grad_output", 2}, in slow_conv_transpose2d_acc_grad_parameters_cuda_template() local
DMaxUnpooling.cu412 grad_output_arg{grad_output_, "grad_output_", 2}, in max_unpooling2d_backward_out_cuda() local
522 grad_output_arg{grad_output_, "grad_output_", 3}, in max_unpooling3d_backward_out_cuda() local
DUpSampleNearest1d.cu154 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_nearest1d_backward_out_cuda_template() local
DUpSampleLinear1d.cu168 TensorArg grad_output_arg{grad_output_, "grad_output_", 1}, in upsample_linear1d_backward_out_cuda_template() local
DNaiveConvolutionTranspose3d.cu449 grad_output_arg{grad_output_, "grad_output", 2}, in slow_conv_transpose3d_backward_out_cuda_template() local
657 grad_output_arg{grad_output_, "grad_output", 2}, in slow_conv_transpose3d_acc_grad_parameters_cuda() local
DUpSampleBicubic2d.cu231 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_bicubic2d_backward_out_cuda_template() local
DUpSampleNearest3d.cu231 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_nearest3d_backward_out_cuda_template() local
DNaiveDilatedConvolution.cu158 bias_arg{bias, "bias", 4}, grad_output_arg{grad_output, "grad_output", 5}; in slow_conv_dilated_location_check() local
DUpSampleTrilinear3d.cu308 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_trilinear3d_backward_out_cuda_template() local
DAdaptiveMaxPooling2d.cu332 TensorArg grad_output_arg{gradOutput, "gradOutput", 2}; in TORCH_IMPL_FUNC() local
DUpSampleBilinear2d.cu368 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_bilinear2d_backward_out_cuda_template() local
780 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_gen2d_aa_backward_out_cuda_template() local
DUpSampleNearest2d.cu343 grad_output_arg{grad_output_, "grad_output_", 2}; in upsample_nearest2d_backward_out_cuda_template() local
DAdaptiveMaxPooling3d.cu391 TensorArg grad_output_arg{gradOutput, "gradOutput", 2}; in TORCH_IMPL_FUNC() local
DAdaptiveAveragePooling3d.cu427 TensorArg grad_output_arg{gradOutput_, "gradOutput_", 2}; in adaptive_avg_pool3d_backward_out_cuda_template() local
DAdaptiveAveragePooling.cu607 grad_output_arg{ gradOutput_, "gradOutput_", 2 }, in adaptive_avg_pool2d_backward_out_cuda_template() local