Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
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() argument
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()
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()
DCUDAJitLoops.cuh136 bool vectorized = vec_size > 1; in launch_jitted_vectorized_kernel() local
/external/pytorch/aten/src/ATen/test/
Dcuda_vectorized_test.cu85 using vectorized = policies::vectorized<vec_size, array_t>; in vectorized_copy() typedef
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorCostModel.h64 bool vectorized, double packet_size) in TensorOpCost()
/external/pytorch/aten/src/ATen/cuda/
Djiterator.cu24 bool vectorized = vec_size > 1; in launch_jitted_vectorized_kernel_dynamic() local
/external/mesa3d/src/freedreno/ir3/
Dir3_context.c115 bool vectorized = false; in ir3_context_init() local