Home
last modified time | relevance | path

Searched defs:grad_refl (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.h184 scalar_t grad_clip, grad_refl; in grid_sampler_compute_source_index_set_grad() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cuh196 scalar_t grad_clip, grad_refl; in grid_sampler_compute_source_index_set_grad() local