Searched refs:thread_reduce_impl (Results 1 – 1 of 1) sorted by relevance
493 return thread_reduce_impl<output_vec_size>(data, [](index_t idx) { return idx; }); in thread_reduce()495 …return thread_reduce_impl<output_vec_size>(data, [&](index_t idx) { return idx * element_stride; }… in thread_reduce()497 …return thread_reduce_impl<output_vec_size>(data, [&](index_t idx) { return input_calc.get(idx)[0] … in thread_reduce()564 …C10_DEVICE at::detail::Array<arg_t, output_vec_size> thread_reduce_impl(const scalar_t* data_, off… in thread_reduce_impl() function