mir_interpreter::HardSwishImpl< T > Struct Template Reference

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

Detailed Description

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

Definition at line 23 of file HardSwish.cpp.

Member Function Documentation

run()

template<typename T >

void mir_interpreter::HardSwishImpl< T >::run ( const mir::TensorVariant &  input, mir::TensorVariant &  result  )

static

Definition at line 29 of file HardSwish.cpp.

{
  auto output_data = reinterpret_cast<T *>(result.atOffset(0));
  auto input_data = reinterpret_cast<T *>(input.atOffset(0));
  auto in_end = input_data + input.getShape().numElements();
  for (; input_data < in_end; input_data++, output_data++)
  {
    const auto in = *input_data;
    *output_data = in * std::min<T>(6.f, std::max<T>(0.f, in + 3.f)) / 6.f;
  }
}

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

---

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

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