Home
last modified time | relevance | path

Searched defs:gradInput_data (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling3d.cu204 scalar_t *gradInput_data, in adaptivemaxgradinput_loop()
280 scalar_t *gradInput_data, in atomicadaptivemaxgradinput_loop()
447 scalar_t* gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
470 scalar_t* gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling3d.cu218 scalar_t *gradInput_data, const scalar_t *gradOutput_data, in adaptiveaveragegradinput_loop()
314 scalar_t* gradInput_data, const scalar_t* gradOutput_data, in atomicadaptiveaveragegradinput_loop()
481 scalar_t* gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cuda_template() local
495 scalar_t* gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cuda_template() local
H A DAdaptiveMaxPooling2d.cu369 scalar_t* gradInput_data = gradInput_c.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
430 scalar_t* gradInput_data = gradInput_c.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAveragePool2d.cu414 scalar_t *gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool2d.cu488 scalar_t *gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling.cu729 scalar_t *gradInput_data = gradInput.mutable_data_ptr<scalar_t>(); in adaptive_avg_pool2d_backward_out_cuda_template() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveMaxPooling3d.cpp233 scalar_t *gradInput_data, in adaptive_max_pool3d_backward_out_frame()
396 scalar_t* gradInput_data = gradInput.data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
416 scalar_t* gradInput_data = gradInput.data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling3d.cpp253 scalar_t* gradInput_data = gradInput.data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cpu_template() local
273 scalar_t* gradInput_data = gradInput.data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cpu_template() local
H A DAveragePool3d.cpp471 scalar_t *gradInput_data = gradInput.data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
495 scalar_t *gradInput_data = gradInput.data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool2d.cpp380 auto gradInput_data = gradInput.data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DSoftMax.cpp295 scalar_t* gradInput_data = in host_softmax_backward() local