mir_interpreter::SubImpl< T > Struct Template Reference

Static Public Member Functions
static void	run (const TensorVariant &lhs, const TensorVariant &rhs, TensorVariant &res)

Detailed Description

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

Definition at line 28 of file Sub.cpp.

Member Function Documentation

run()

template<typename T >

void mir_interpreter::SubImpl< T >::run ( const TensorVariant &  lhs, const TensorVariant &  rhs, TensorVariant &  res  )

static

Definition at line 34 of file Sub.cpp.

{
  TensorVariant broadcasted_lhs(lhs, res.getShape());
  TensorVariant broadcasted_rhs(rhs, res.getShape());
  Tensor<T> lhs_accessor(broadcasted_lhs);
  Tensor<T> rhs_accessor(broadcasted_rhs);
  Tensor<T> res_accessor(res);
 
  for (const auto &index : ShapeRange(res.getShape()))
  {
    res_accessor.at(index) = lhs_accessor.at(index) - rhs_accessor.at(index);
  }
}

References mir::Tensor< T >::at(), and mir::TensorVariant::getShape().

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

---

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

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