Method shape

compiler/loco/src/IR/PermutingCodec.cpp:109–121

Permuting Encoder

// Permuting Encoder
//
FeatureShape PermutingEncoder<Domain::Feature>::shape(const TensorShape &in) const
{
  assert(valid() && "invalid permutation");

  FeatureShape out;

  out.count() = in.dim(_perm[FeatureAxis::Count]);
  out.depth() = in.dim(_perm[FeatureAxis::Depth]);
  out.height() = in.dim(_perm[FeatureAxis::Height]);
  out.width() = in.dim(_perm[FeatureAxis::Width]);

  return out;
}

TensorIndex PermutingEncoder<Domain::Feature>::value(const FeatureIndex &in) const
{