Home
last modified time | relevance | path

Searched defs:vectorized (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DMemoryAccess.cuh241 struct vectorized { struct
244 static constexpr int loop_size = thread_work_size() / vec_size;
246 data_t data;
248 __device__ vectorized(data_t data) : data(data) {} in vectorized() function
250 __device__ inline constexpr bool check_inbounds(int thread_work_elem) { in check_inbounds()
255 __device__ inline void load_single_arg(accessor_t to, scalar_t *from) { in load_single_arg()
269 __device__ inline void load(args_t *args, int idx) { in load()
275 __device__ inline void store(scalar_t *from, int idx) { in store()
H A Djit_utils.cpp941 bool vectorized, in generate_code()
982 bool vectorized, in generate_code()
1308 bool vectorized, in generate_reduction_code()
1338 bool vectorized, in generate_reduction_code()
H A DCUDAJitLoops.cuh136 bool vectorized = vec_size > 1; in launch_jitted_vectorized_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcuda_vectorized_test.cu85 using vectorized = policies::vectorized<vec_size, array_t>; in vectorized_copy() typedef
/aosp_15_r20/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
H A DTensorCostModel.h64 bool vectorized, double packet_size) in TensorOpCost()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu24 bool vectorized = vec_size > 1; in launch_jitted_vectorized_kernel_dynamic() local
/aosp_15_r20/external/mesa3d/src/freedreno/ir3/
H A Dir3_context.c106 bool vectorized = false; in ir3_context_init() local