const ::internal::tflite::operand::Index output_index{node.param().output_index};
const ::internal::tflite::operand::Index input_index{node.param().input_index};
- // TODO Re-enable this assert
- // assert((ifm_shape.C * ifm_shape.H * ifm_shape.W) == out_size);
-
- // TODO Should move to the place where the operand is handled, if it is possible.
-
auto input_shape = asTensorShape(_ctx.at(input_index).shape());
auto output_shape = asTensorShape(_ctx.at(output_index).shape());
assert(input_shape[0] * input_shape[1] * input_shape[2] * input_shape[3] ==
output_shape[0] * output_shape[1] * output_shape[2] * output_shape[3]);
+ // TODO Should move to the place where the operand is handled, if it is possible.
_builder.addShapeConstr(output_index, asTensorInfo(output_shape, _ctx.at(output_index).type(),
_ctx.at(output_index).scale(),
_ctx.at(output_index).zeroPoint()));