mir_interpreter::SliceImpl< T > Struct Template Reference

Static Public Member Functions
static void	run (const mir::TensorVariant &arg, const mir::Shape &starts, mir::TensorVariant &res)

Detailed Description

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

Definition at line 28 of file Slice.cpp.

Member Function Documentation

run()

template<typename T >

void mir_interpreter::SliceImpl< T >::run ( const mir::TensorVariant &  arg, const mir::Shape &  starts, mir::TensorVariant &  res  )

static

Definition at line 34 of file Slice.cpp.

{
  mir::Tensor<T> input(arg);
  mir::Tensor<T> output(res);
 
  for (auto id : mir::ShapeRange(res.getShape()))
  {
    mir::Index idx = mir_interpreter::shift(id, starts);
    output.at(id) = input.at(idx);
  }
}

References mir::TensorVariant::getShape(), and mir_interpreter::shift().

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

---

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

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