ternary_op Class — pytorch Architecture

Source Code

aten/src/ATen/cpu/vec/vec_n.h lines 80–93

  template <typename Op>
  inline VectorizedN<T, N> ternary_op(
      const VectorizedN<T, N>& other,
      const VectorizedN<T, N>& other2,
      Op op) const {
    VectorizedN<T, N> result;
#ifndef _MSC_VER
#pragma unroll
#endif
    for (int i = 0; i < N; ++i) {
      result.values[i] = op(values[i], other.values[i], other2.values[i]);
    }
    return result;
  }