mir_interpreter::CappedReLUImpl< T > Struct Template Reference

Static Public Member Functions
static void	run (const mir::TensorVariant &arg, float cap, mir::TensorVariant &result)

Detailed Description

template<typename T>
struct mir_interpreter::CappedReLUImpl< T >

Definition at line 30 of file CappedReLU.cpp.

Member Function Documentation

run()

template<typename T >

void mir_interpreter::CappedReLUImpl< T >::run ( const mir::TensorVariant &  arg, float  cap, mir::TensorVariant &  result  )

static

Definition at line 36 of file CappedReLU.cpp.

{
  mir::Tensor<T> arg_accessor(arg);
  mir::Tensor<T> res_accessor(result);
 
  for (const auto &index : mir::ShapeRange(result.getShape()))
  {
    res_accessor.at(index) = std::min(std::max(arg_accessor.at(index), T(0)), static_cast<T>(cap));
  }
}

References mir::Tensor< T >::at().

Referenced by package.infer.session::inference().

---

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

• compiler/mir-interpreter/src/ops/CappedReLU.cpp