* Checked that resize work only H and W dims
Signed-off-by: Pavel Iliutchenko <p.iliutchenk@samsung.com>
void ModelAnalyzer::visit(mir::ops::ResizeOp &op)
{
+ const auto &in_shape = op.getInputShape(0);
+ const auto &out_shape = op.getOutputShape(0);
+
+ assert(in_shape.rank() == 4);
+ assert(in_shape.rank() == out_shape.rank());
+
+ if (in_shape.dim(0) != out_shape.dim(0) || in_shape.dim(3) != out_shape.dim(3))
+ throw std::runtime_error("Not supported Resize on other dims besides height and width!");
+
switch (op.getMode())
{
case mir::ops::ResizeOp::ResizeMethod::nearestNeighbor: