Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cpu/
DUpSampleKernel.cpp1040 static const int interp_size = 1; member
1156 static const int interp_size = 2; member
1236 auto interp_size = HelperInterpLinear::interp_size; in compute_index_ranges_weights() local
1266 auto interp_size = HelperInterpLinear::interp_size; in compute_index_ranges_int16_weights() local
1276 static const int interp_size = 4; member
1369 auto interp_size = HelperInterpCubic::interp_size; in compute_index_ranges_weights() local
1399 auto interp_size = HelperInterpCubic::interp_size; in compute_index_ranges_int16_weights() local
1443 constexpr int interp_size = F::interp_size; in upsample_generic_Nd_kernel_impl() local
1949 int interp_size = F::interp_size; in cpu_upsample_genNd_backward_aa() local
/external/pytorch/aten/src/ATen/native/cuda/
DUpSample.cuh326 int interp_size, in _compute_weights()