Home
last modified time | relevance | path

Searched refs:basic_loop (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DLoops.h115 basic_loop(char* C10_RESTRICT data[], const int64_t* strides_, int64_t i, int64_t n, func_t&& op) { in basic_loop() function
225 basic_loop(data, strides, i, n, std::forward<func_t>(op));
288 basic_loop(data.data(), strides, 0, size0, op);
315 basic_loop(data, strides, 0, n, op);
366 basic_loop(data, strides, 0, n, op);
H A DReduce.h91 basic_loop(ptrs, strides, count * 4 * Vec::size(), n, op); in vectorized_inner_reduction()
115 basic_loop(ptrs, strides, 0, size0, op); in vectorized_outer_reduction()
280 basic_loop(ptrs, inner_strides, 0, size0, op);
H A DSumKernel.cpp568 basic_loop(ptrs, inner_strides, 0, size0, [](scalar_t a, scalar_t b) { in cascade_sum()
574 basic_loop(ptrs, inner_strides, 0, size0, in cascade_sum()
H A DUpSampleKernel.cpp291 static inline void basic_loop(char** data, const int64_t* strides, int64_t n) { in basic_loop() function
445 basic_loop<scalar_t, int64_t, out_ndims, interp_size>(data, strides, n); in cpu_upsample_generic()
449 basic_loop<scalar_t, int64_t, out_ndims, interp_size>(data, strides, n); in cpu_upsample_generic()
452 basic_loop<scalar_t, int64_t, out_ndims, interp_size>(data, strides, n); in cpu_upsample_generic()