MCPcopy Create free account
hub / github.com/ARM-software/armnn / SetNumDimensions

Method SetNumDimensions

src/armnn/Tensor.cpp:219–230  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

217}
218
219void TensorShape::SetNumDimensions(unsigned int numDimensions, bool initDimensionsSpecificity)
220{
221 CheckScalar();
222 CheckSpecifiedNumDimensions();
223 CheckValidNumDimensions(numDimensions);
224
225 m_NumDimensions = numDimensions;
226 m_Dimensionality = Dimensionality::Specified;
227 std::fill(m_Dimensions.begin(), m_Dimensions.begin() + m_NumDimensions, 0);
228 std::fill(m_DimensionsSpecificity.begin(), m_DimensionsSpecificity.begin() + m_NumDimensions,
229 initDimensionsSpecificity);
230}
231
232void TensorShape::SetDimensionSize(unsigned int i, unsigned int dimensionSize)
233{

Callers 3

VisitUnpackOperatorFunction · 0.80
VisitUnpackOperatorFunction · 0.80
TensorTest.cppFile · 0.80

Calls 2

fillFunction · 0.85
beginMethod · 0.45

Tested by

no test coverage detected