This patch adds necessary assertions in TransposeConv.
Signed-off-by: b.devansh <b.devansh@samsung.com>
const PaddingCode padding_type =
static_cast<PaddingCode>(_ctx.at(padding_index).asScalar<int32_t>());
+ assert(vstride > 0);
+ assert(hstride > 0);
+ assert((ANEURALNETWORKS_PADDING_SAME == padding_type) ||
+ (ANEURALNETWORKS_PADDING_VALID == padding_type));
+ assert(ifm_shape.N == ofm_shape.N);
+ assert(ifm_shape.C == ker_shape.C);
+ assert(ker_shape.N == ofm_shape.C);
+
// Set Shape Constraints and TensorInfo
_builder.addShapeConstr(
ifm_index, asTensorInfo(asTensorShape(_ctx.at(ifm_index).shape()), _ctx.at(ifm_index).type(),