Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling3d.cu205 const scalar_t *gradOutput_data, in adaptivemaxgradinput_loop()
281 const scalar_t *gradOutput_data, in atomicadaptivemaxgradinput_loop()
448 const scalar_t* gradOutput_data = gradOutput_.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
471 const scalar_t* gradOutput_data = gradOutput_.const_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()
482 const scalar_t* gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cuda_template() local
496 const scalar_t* gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cuda_template() local
H A DAdaptiveMaxPooling2d.cu370 const scalar_t* gradOutput_data = gradOutput_.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
431 const scalar_t* gradOutput_data = gradOutput_.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAveragePool2d.cu413 const scalar_t *gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool2d.cu487 const scalar_t *gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling.cu728 const scalar_t *gradOutput_data = gradOutput.const_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.cpp234 const scalar_t *gradOutput_data, in adaptive_max_pool3d_backward_out_frame()
397 const scalar_t* gradOutput_data = gradOutput_.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
417 const scalar_t* gradOutput_data = gradOutput_.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveAveragePooling3d.cpp254 const scalar_t* gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cpu_template() local
274 const scalar_t* gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in adaptive_avg_pool3d_backward_out_cpu_template() local
H A DAveragePool3d.cpp472 const scalar_t *gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
496 const scalar_t *gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DFractionalMaxPool2d.cpp381 auto gradOutput_data = gradOutput.const_data_ptr<scalar_t>(); in TORCH_IMPL_FUNC() local
H A DSoftMax.cpp299 const scalar_t* gradOutput_data = in host_softmax_backward() local