diff --git a/include/aidge/operator/Resize.hpp b/include/aidge/operator/Resize.hpp index f66588746a30280dd606e4ef3f196cec50b11f7e..3ecf74d2b1efcdc802d2010f879293663fbb9041 100644 --- a/include/aidge/operator/Resize.hpp +++ b/include/aidge/operator/Resize.hpp @@ -98,15 +98,6 @@ inline std::shared_ptr<Node> Resize(const std::size_t nbInputDims, const bool noSizes, const std::string &name = "") { - fmt::print("noROI {}\n", noROI); - fmt::print("IF {}\n", std::array<DimSize_t, 1>({noROI ? 0 : nbInputDims})); - - fmt::print("noScales {}\n", noScales); - fmt::print("IF {}\n", std::array<DimSize_t, 1>({noScales ? 0 : nbInputDims})); - - fmt::print("noSizes {}\n", noSizes); - fmt::print("IF {}\n", std::array<DimSize_t, 1>({noSizes ? 0 : nbInputDims})); - auto resize_node = std::make_shared<Node>(std::make_shared<Resize_Op>(noROI, noScales, noSizes), name); // create empty producers of the same as the rank of input size [nbInputDims] diff --git a/src/operator/Resize.cpp b/src/operator/Resize.cpp index 89bf9965d6fbdb697a9933d5f81f4794ca1ff57c..7a85588f558b98384d5054b19eeb08dd08a32ea7 100644 --- a/src/operator/Resize.cpp +++ b/src/operator/Resize.cpp @@ -28,6 +28,9 @@ const std::string Aidge::Resize_Op::Type = "Resize"; bool Aidge::Resize_Op::forwardDims(bool /*allowDataDependency*/) { + AIDGE_ASSERT(getInput(0)->nbDims() == 4,\ + "input tensor must have dimentions = 4 ."); + // check input ("data_input","roi", "scales", "data_input","sizes") has been associated for (size_t i = 0; i < 4; ++i) { if (!getInput(i)) {