Home
last modified time | relevance | path

Searched defs:gix (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DGridSampler.cu387 scalar_t gix = static_cast<scalar_t>(0), giy = static_cast<scalar_t>(0); in grid_sampler_2d_backward_kernel() local
474 scalar_t gix = static_cast<scalar_t>(0); in grid_sampler_2d_backward_kernel() local
632 …scalar_t gix = static_cast<scalar_t>(0), giy = static_cast<scalar_t>(0), giz = static_cast<scalar_… in grid_sampler_3d_backward_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DGridSampler.cpp346 …scalar_t gix = static_cast<scalar_t>(0), giy = static_cast<scalar_t>(0), giz = static_cast<scalar_… in grid_sampler_3d_backward_cpu_impl() local
806 scalar_t gix = static_cast<scalar_t>(0), giy = static_cast<scalar_t>(0); in _grid_sampler_2d_cpu_fallback_backward() local
883 scalar_t gix = static_cast<scalar_t>(0); in _grid_sampler_2d_cpu_fallback_backward() local
/aosp_15_r20/external/tensorflow/tensorflow/core/util/sparse/
H A Dsparse_tensor_test.cc522 for (std::size_t gix = 0; gix < groups.size(); ++gix) { in TEST() local