Home
last modified time | relevance | path

Searched refs:cudnn_grid_sampler_backward (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/cudnn/
DGridSampler.cpp27 std::tuple<Tensor, Tensor> cudnn_grid_sampler_backward( in cudnn_grid_sampler_backward() function
127 std::tuple<Tensor, Tensor> cudnn_grid_sampler_backward( in cudnn_grid_sampler_backward() function
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesModules.cpp360 …VMAP_SUPPORT(cudnn_grid_sampler_backward, CUDNN_GRID_SAMPLE_BW_BATCH_RULE(cudnn_grid_sampler_backw… in TORCH_LIBRARY_IMPL()
/external/pytorch/test/expect/
DHasDecompTest.test_has_decomposition.expect756 aten::cudnn_grid_sampler_backward
757 aten::cudnn_grid_sampler_backward.out