clamp Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec256/vec256_int.h lines 1353–1369

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