Home
last modified time | relevance | path

Searched defs:min_vec (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DTensorCompareKernel.cpp361 const Vectorized<scalar_t> min_vec(min); in clamp_scalar_kernel_impl() local
390 const Vectorized<scalar_t> min_vec(min); in clamp_min_scalar_kernel_impl() local
H A DActivation.cpp40 Vectorized<float> min_vec = minimum(data_vec0, Vectorized<float>(float(0))); in log_sigmoid_cpu_kernel() local
52 Vectorized<float> min_vec = minimum(data_vec0, Vectorized<float>(float(0))); in log_sigmoid_cpu_kernel() local
74 Vec min_vec = vec::minimum(data_vec, Vec(scalar_t(0))); in log_sigmoid_cpu_kernel() local
82 Vec min_vec = vec::minimum(data_vec, Vec(scalar_t(0))); in log_sigmoid_cpu_kernel() local
H A DSoftMaxKernel.cpp921 Vec min_vec = Vec(-std::numeric_limits<scalar_t>::infinity()); in _vec_logsoftmax() local
H A DBinaryOpsKernel.cpp49 auto min_vec = Vectorized<scalar_t>(min_scalar); in add_clamp_kernel() local
/aosp_15_r20/external/executorch/kernels/optimized/vec/
H A Dvec_base.h499 Vectorized<T> inline clamp(const Vectorized<T> &a, const Vectorized<T> &min_vec, const Vectorized<T… in clamp()
517 Vectorized<T> inline clamp_min(const Vectorized<T> &a, const Vectorized<T> &min_vec) { in clamp_min()
/aosp_15_r20/external/pytorch/test/cpp/aoti_abi_check/
H A Dtest_vec.cpp71 Vec min_vec = Vec::loadu(min.data()); in TEST() local
/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()
504 const auto min_vec = wrapper::vdup_n(static_cast<T>(min_bound), ExactTagType{}); in run_offset_contribution_output_stage() local
/aosp_15_r20/external/pytorch/aten/src/ATen/cpu/vec/
H A Dvec_base.h739 Vectorized<T> inline clamp(const Vectorized<T> &a, const Vectorized<T> &min_vec, const Vectorized<T… in clamp()
759 Vectorized<T> inline clamp_min(const Vectorized<T> &a, const Vectorized<T> &min_vec) { in clamp_min()
/aosp_15_r20/external/libvpx/vpx_dsp/mips/
H A Dmacros_msa.h730 #define MIN_UH2(RTYPE, in0, in1, min_vec) \ argument
737 #define MIN_UH4(RTYPE, in0, in1, in2, in3, min_vec) \ argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp956 auto min_vec = Vec(min_q); in qclamp_kernel() local
987 auto min_vec = Vec(min_q); in qclamp_min_kernel() local
/aosp_15_r20/external/vixl/test/aarch64/
H A Dtest-assembler-neon-aarch64.cc10559 uint64_t min_vec = Float16ToV4H(min); in FminFmaxFloat16Helper() local