Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.h57 inline int can_vectorize_up_to(size_t default_alignment, void *pointer) { in can_vectorize_up_to() function
68 inline int can_vectorize_up_to(const KernelDescriptor &desc, c10::ArrayRef<char*> pointers) { in can_vectorize_up_to() function
H A DMemoryAccess.cuh349 inline C10_HOST_DEVICE int can_vectorize_up_to(const char *pointer) { in can_vectorize_up_to() function
362 inline C10_HOST_DEVICE int can_vectorize_up_to(char *pointer) { in can_vectorize_up_to() function
378 inline int can_vectorize_up_to(array_t pointers) { in can_vectorize_up_to() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator_impl.h47 int can_vectorize_up_to(at::ScalarType type, char* pointer) { in can_vectorize_up_to() function