Home
last modified time | relevance | path

Searched defs:max_vec (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_opt_vectorize.c52 uint32_t num_components, uint32_t max_vec) in hash_alu_src()
67 uint32_t max_vec) in hash_phi_src()
134 uint32_t max_vec) in alu_srcs_equal()
145 const nir_phi_src *src2, uint32_t max_vec) in phi_srcs_equal()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp823 Vec max_vec = Vec::loadu(input_data); in _vec_softmax() local
940 Vec max_vec = Vec::loadu(input_max_data + d1); in _vec_logsoftmax() local
960 Vec max_vec = Vec::loadu(input_max_data + d2); in _vec_logsoftmax() local
983 Vec max_vec = Vec::loadu(input_max_data + d3); in _vec_logsoftmax() local
H A DTensorCompareKernel.cpp362 const Vectorized<scalar_t> max_vec(max); in clamp_scalar_kernel_impl() local
376 const Vectorized<scalar_t> max_vec(max); in clamp_max_scalar_kernel_impl() local
H A DBinaryOpsKernel.cpp51 auto max_vec = Vectorized<scalar_t>(max_scalar); in add_clamp_kernel() local
/aosp_15_r20/external/executorch/kernels/optimized/vec/
H A Dvec_base.h499 … inline clamp(const Vectorized<T> &a, const Vectorized<T> &min_vec, const Vectorized<T> &max_vec) { in clamp()
508 Vectorized<T> inline clamp_max(const Vectorized<T> &a, const Vectorized<T> &max_vec) { in clamp_max()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuGemmLowpOffsetContributionOutputStageKernel.cpp310 … typename VT::vtype min_vec, typename VT::vtype max_vec, in run_offset_contribution_output_stage_window()
505 const auto max_vec = wrapper::vdup_n(static_cast<T>(max_bound), ExactTagType{}); in run_offset_contribution_output_stage() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/
H A Dvec_base.h739 … inline clamp(const Vectorized<T> &a, const Vectorized<T> &min_vec, const Vectorized<T> &max_vec) { in clamp()
749 Vectorized<T> inline clamp_max(const Vectorized<T> &a, const Vectorized<T> &max_vec) { in clamp_max()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp957 auto max_vec = Vec(max_q); in qclamp_kernel() local
1012 auto max_vec = Vec(max_q); in qclamp_max_kernel() local
/aosp_15_r20/external/vixl/test/aarch64/
H A Dtest-assembler-neon-aarch64.cc10560 uint64_t max_vec = Float16ToV4H(max); in FminFmaxFloat16Helper() local