Home
last modified time | relevance | path

Searched refs:jitted_can_vectorize_up_to (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/cuda/
Djiterator_impl.h61 int jitted_can_vectorize_up_to(const TensorIteratorBase& iter) { in jitted_can_vectorize_up_to() function
Djiterator.cu23 const int vec_size = jitted_can_vectorize_up_to(iter); in launch_jitted_vectorized_kernel_dynamic()