Searched defs:input_requires_grad (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cpp | 48 auto input_requires_grad = output_mask[0]; in grid_sampler_2d_backward_cuda() local
67 auto input_requires_grad = output_mask[0]; in grid_sampler_3d_backward_cuda() local
|
| H A D | GridSampler.cu | 323 const bool input_requires_grad) { in grid_sampler_2d_backward_kernel()
529 const bool input_requires_grad) { in grid_sampler_3d_backward_kernel()
859 auto input_requires_grad = output_mask[0]; in launch_grid_sampler_2d_backward_kernel() local
920 auto input_requires_grad = output_mask[0]; in launch_grid_sampler_3d_backward_kernel() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | GridSamplerKernel.cpp | 1230 auto input_requires_grad = output_mask[0]; in grid_sampler_2d_backward_cpu_kernel_impl() local
1239 #define GINP_SLICE_PTR(input_requires_grad) GINP_SLICE_PTR_##input_requires_grad in grid_sampler_2d_backward_cpu_kernel_impl() argument
1241 #define HANDLE_CASE(interp, padding, align_corners, input_requires_grad) \ in grid_sampler_2d_backward_cpu_kernel_impl() argument
1265 #define HANDLE_INTERP(interp, align_corners, input_requires_grad) \ in grid_sampler_2d_backward_cpu_kernel_impl() argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | sdp_utils_cpp.h | 67 inline bool input_requires_grad(sdp_params const& params) { in input_requires_grad() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | GridSampler.cpp | 216 auto input_requires_grad = output_mask[0]; in grid_sampler_3d_backward_cpu_impl() local
1010 auto input_requires_grad = output_mask[0]; in grid_sampler_2d_backward_cpu() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 5533 def test(N, C, H, W, mode, padding_mode, align_corners, input_requires_grad): argument
5894 def test(N, C, D, H, W, mode, padding_mode, align_corners, input_requires_grad): argument
|
| H A D | test_autograd.py | 7075 def test_checkpointing_without_reentrant(self, input_requires_grad): argument
9084 def get_ref(input_requires_grad, nb_hooks): argument
|
| H A D | test_mps.py | 10606 def test(N, C, H, W, mode, padding_mode, align_corners, input_requires_grad): argument
|