nnfw::cker::bias_op::functor::Bias< Device, T > Struct Template Reference

#include <bias_op.h>

Public Member Functions
void	operator() (const Device &d, typename TTypes< T >::ConstFlat input, typename TTypes< T >::ConstVec bias, typename TTypes< T >::Flat output, T activation_min, T activation_max)

Detailed Description

template<typename Device, typename T>
struct nnfw::cker::bias_op::functor::Bias< Device, T >

Definition at line 59 of file bias_op.h.

Member Function Documentation

operator()()

template<typename Device , typename T >

void nnfw::cker::bias_op::functor::Bias< Device, T >::operator() ( const Device &  d, typename TTypes< T >::ConstFlat  input, typename TTypes< T >::ConstVec  bias, typename TTypes< T >::Flat  output, T  activation_min, T  activation_max  )

inline

Definition at line 62 of file bias_op.h.

  {
    const Eigen::Index rest_size = input.size() / bias.dimension(0);
    Eigen::DSizes<Eigen::Index, 1> bcast(rest_size);
    MaybeWith32BitIndexing<Device>(
      [&](auto input32, auto bias32, typename TTypes<T>::Flat output32, const auto &bcast32,
          T activation_min, T activation_max) {
        output32.device(d) =
          (input32 + bias32.broadcast(bcast32))
            .template cwiseMax<Eigen::PropagateNaN>(static_cast<T>(activation_min))
            .template cwiseMin<Eigen::PropagateNaN>(static_cast<T>(activation_max));
      },
      input, bias, output, bcast, activation_min, activation_max);
  }

---

The documentation for this struct was generated from the following file:

• compute/cker/include/cker/eigen/bias_op.h