Searched refs:dimensionSizes (Results 1 – 4 of 4) sorted by relevance
| /external/armnn/src/armnn/ |
| D | Tensor.cpp | 40 TensorShape::TensorShape(const unsigned int numDimensions, const unsigned int* const dimensionSizes) in TensorShape() argument
45 if (dimensionSizes == nullptr) in TensorShape()
50 std::copy(dimensionSizes, dimensionSizes + numDimensions, m_Dimensions.begin()); in TensorShape()
60 const unsigned int* const dimensionSizes, in TensorShape() argument
66 if (dimensionSizes == nullptr) in TensorShape()
76 std::copy(dimensionSizes, dimensionSizes + numDimensions, m_Dimensions.begin()); in TensorShape()
360 const unsigned int* dimensionSizes, in TensorInfo() argument
365 : m_Shape(numDimensions, dimensionSizes), m_DataType(dataType), m_IsConstant(isConstant) in TensorInfo()
385 const unsigned int* dimensionSizes, in TensorInfo() argument
390 : m_Shape(numDimensions, dimensionSizes) in TensorInfo()
|
| /external/armnn/src/armnn/layers/ |
| D | StackLayer.cpp | 45 std::vector<unsigned int> dimensionSizes(inputNumDimensions + 1, 0); in InferOutputShapes() local
48 dimensionSizes[i] = inputShape[i]; in InferOutputShapes()
51 dimensionSizes[axis] = m_Param.m_NumInputs; in InferOutputShapes()
55 dimensionSizes[i] = inputShape[i-1]; in InferOutputShapes()
58 TensorShape targetShape = TensorShape(inputNumDimensions + 1, dimensionSizes.data()); in InferOutputShapes()
|
| /external/armnn/include/armnn/ |
| D | Tensor.hpp | 34 TensorShape(unsigned int numDimensions, const unsigned int* dimensionSizes);
48 …TensorShape(unsigned int numDimensions, const unsigned int* dimensionSizes, const bool* dimensions…
165 const unsigned int* dimensionSizes,
178 const unsigned int* dimensionSizes,
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_tensor.i | 43 %tensor_shape_typemap(unsigned int numDimensions, const unsigned int* dimensionSizes);
44 TensorShape(unsigned int numDimensions, const unsigned int* dimensionSizes);
45 %clear_tensor_shape_typemap(unsigned int numDimensions, const unsigned int* dimensionSizes);
|