diff options
Diffstat (limited to 'src/armnn/layers/BatchToSpaceNdLayer.cpp')
-rw-r--r-- | src/armnn/layers/BatchToSpaceNdLayer.cpp | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/src/armnn/layers/BatchToSpaceNdLayer.cpp b/src/armnn/layers/BatchToSpaceNdLayer.cpp index 7e7045291c..1da88c63ac 100644 --- a/src/armnn/layers/BatchToSpaceNdLayer.cpp +++ b/src/armnn/layers/BatchToSpaceNdLayer.cpp @@ -47,7 +47,7 @@ void BatchToSpaceNdLayer::ValidateTensorShapesFromInputs() auto inferredShapes = InferOutputShapes({GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape()}); - BOOST_ASSERT(inferredShapes.size() == 1); + ARMNN_ASSERT(inferredShapes.size() == 1); ConditionalThrowIfNotEqual<LayerValidationException>( "BatchToSpaceLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.", @@ -56,7 +56,7 @@ void BatchToSpaceNdLayer::ValidateTensorShapesFromInputs() std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const { - BOOST_ASSERT(inputShapes.size() == 1); + ARMNN_ASSERT(inputShapes.size() == 1); const TensorShape& inputShape = inputShapes[0]; TensorShape outputShape(inputShape); @@ -66,7 +66,7 @@ std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vecto 1U, std::multiplies<>()); - BOOST_ASSERT(inputShape[0] % accumulatedBlockShape == 0); + ARMNN_ASSERT(inputShape[0] % accumulatedBlockShape == 0); outputShape[0] = inputShape[0] / accumulatedBlockShape; @@ -80,10 +80,10 @@ std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vecto unsigned int outputHeight = inputShape[heightIndex] * m_Param.m_BlockShape[0]; unsigned int outputWidth = inputShape[widthIndex] * m_Param.m_BlockShape[1]; - BOOST_ASSERT_MSG(heightCrop <= outputHeight, + ARMNN_ASSERT_MSG(heightCrop <= outputHeight, "BatchToSpaceLayer: Overall height crop should be less than or equal to the uncropped output height."); - BOOST_ASSERT_MSG(widthCrop <= outputWidth, + ARMNN_ASSERT_MSG(widthCrop <= outputWidth, "BatchToSpaceLayer: Overall width crop should be less than or equal to the uncropped output width."); outputShape[heightIndex] = outputHeight - heightCrop; |