Searched defs:vec_idx (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NEReductionOperationKernel.cpp | 76 uint32x4_t vec_idx = { idx, idx + 1, idx + 2, idx + 3 }; in calculate_index() local
106 uint32x4x4_t vec_idx = { { { idx + 0, idx + 1, idx + 2, idx + 3 }, in calculate_index_quantized() local
266 uint32x4x2_t vec_idx = { { { idx + 0, idx + 1, idx + 2, idx + 3 }, in calculate_index() local
|
| /external/mesa3d/src/asahi/lib/ |
| D | agx_nir_prolog_epilog.c | 184 unsigned vec_idx = ~0; in agx_nir_vs_prolog() local
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | layer_norm_kernel.cu | 432 unsigned int vec_idx = l / vec_size; in layer_norm_grad_input_kernel_vectorized() local
474 unsigned int vec_idx = l / vec_size; in layer_norm_grad_input_kernel_vectorized() local
|