}
}
- unsigned int dimSizes[outputRank];
- memset(dimSizes, 1, outputRank * sizeof(unsigned int));
-
+ std::vector<unsigned int> dimSizes(outputRank, 1);
if (!m_Param.m_Axis.empty())
{
// Skip the dimension that has been reduced unless keepDims is true.
}
}
}
- const TensorShape& inferredShape = TensorShape(outputRank, dimSizes);
+ const TensorShape& inferredShape = TensorShape(outputRank, dimSizes.data());
ConditionalThrowIfNotEqual<LayerValidationException>(
"MeanLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
inferredShape);
}
-} // namespace armnn
\ No newline at end of file
+} // namespace armnn