clamp_min Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/vec256_int.h lines 1444–1455

template <>
Vectorized<int64_t> inline clamp_min(
    const Vectorized<int64_t>& a,
    const Vectorized<int64_t>& min_val) {
#ifndef CPU_CAPABILITY_AVX2
  return emulate(a, min_val, [](int64_t a_point, int64_t min_point) {
    return std::max(min_point, a_point);
  });
#else
  return maximum(min_val, a);
#endif
}