Home
last modified time | relevance | path

Searched refs:m_DimMappings (Results 1 – 25 of 29) sorted by relevance

12

/external/armnn/include/armnn/
DTypes.hpp346 return m_DimMappings.at(i); in operator []()
351 ConstIterator begin() const { return m_DimMappings.begin(); } in begin()
356 ConstIterator end() const { return m_DimMappings.begin() + m_NumDimMappings; } in end()
363 if (m_DimMappings[i] != other.m_DimMappings[i]) return false; in IsEqual()
373 isInverse = (m_DimMappings[other.m_DimMappings[i]] == i); in IsInverse()
379 ArrayType m_DimMappings; member in armnn::PermutationVector
DDescriptors.hpp152 : m_DimMappings{} in PermuteDescriptor()
156 : m_DimMappings(dimMappings) in PermuteDescriptor()
161 return m_DimMappings.IsEqual(rhs.m_DimMappings); in operator ==()
173 PermutationVector m_DimMappings; member
1460 : m_DimMappings{} in TransposeDescriptor()
1464 : m_DimMappings(dimMappings) in TransposeDescriptor()
1469 return m_DimMappings.IsEqual(rhs.m_DimMappings); in operator ==()
1481 PermutationVector m_DimMappings; member
/external/armnn/python/pyarmnn/test/
Dtest_descriptors.py263 assert desc.m_DimMappings.GetSize() == 4
264 assert desc.m_DimMappings[0] == 0
265 assert desc.m_DimMappings[1] == 2
266 assert desc.m_DimMappings[2] == 3
267 assert desc.m_DimMappings[3] == 1
443 assert desc.m_DimMappings.GetSize() == 5
444 assert desc.m_DimMappings[0] == 0
445 assert desc.m_DimMappings[1] == 3
446 assert desc.m_DimMappings[2] == 2
447 assert desc.m_DimMappings[3] == 1
[all …]
/external/armnn/src/backends/neon/workloads/
DNeonTransposeWorkload.cpp21 const armnn::PermutationVector& mappings = descriptor.m_DimMappings; in NeonTransposeWorkloadValidate()
41 const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in NeonTransposeWorkload()
DNeonPermuteWorkload.cpp21 const armnn::PermutationVector& mappings = descriptor.m_DimMappings; in NeonPermuteWorkloadValidate()
43 const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in NeonPermuteWorkload()
/external/armnn/src/backends/cl/workloads/
DClTransposeWorkload.cpp23 const armnn::PermutationVector& mappings = descriptor.m_DimMappings; in ClTransposeWorkloadValidate()
44 const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in ClTransposeWorkload()
DClPermuteWorkload.cpp23 const armnn::PermutationVector& mappings = descriptor.m_DimMappings; in ClPermuteWorkloadValidate()
46 const armnn::PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in ClPermuteWorkload()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DPermuteTestImpl.hpp72 descriptor.m_DimMappings = {0U, 3U, 1U, 2U}; in SimplePermuteTest()
122 descriptor.m_DimMappings = {0U, 2U, 3U, 1U}; in PermuteValueSet1Test()
173 descriptor.m_DimMappings = {0U, 3U, 1U, 2U}; in PermuteValueSet2Test()
224 descriptor.m_DimMappings = {0U, 2U, 3U, 1U}; in PermuteValueSet3Test()
DTransposeTestImpl.hpp74 descriptor.m_DimMappings = {0U, 2U, 3U, 1U}; in SimpleTransposeTest()
124 descriptor.m_DimMappings = {0U, 3U, 1U, 2U}; in TransposeValueSet1Test()
175 descriptor.m_DimMappings = {0U, 2U, 3U, 1U}; in TransposeValueSet2Test()
226 descriptor.m_DimMappings = {0U, 3U, 1U, 2U}; in TransposeValueSet3Test()
/external/armnn/src/armnn/test/
DUtilsTests.cpp234 CHECK_THROWS_AS(desc.m_DimMappings[3], armnn::InvalidArgumentException);
235 CHECK(desc.m_DimMappings[0] == 1u);
242 CHECK_THROWS_AS(desc.m_DimMappings[3], armnn::InvalidArgumentException);
243 CHECK(desc.m_DimMappings[2] == 0u);
DShapeInferenceTests.cpp508 descriptor.m_DimMappings = {0U, 2U, 3U, 1U};
730 descriptor.m_DimMappings = {0U, 3U, 1U, 2U};
/external/armnn/src/backends/tosaCommon/operatorMappings/
DTransposeOperator.cpp29 std::vector<int32_t> mappings(transposeDescriptor->m_DimMappings.begin(), in ConvertTransposeToTosaOperator()
30 transposeDescriptor->m_DimMappings.end()); in ConvertTransposeToTosaOperator()
/external/armnn/src/armnn/layers/
DPermuteLayer.hpp40 return m_Param.m_DimMappings; in GetPermutation()
DTransposeLayer.hpp39 return m_Param.m_DimMappings; in GetPermutation()
DPermuteLayer.cpp42 return std::vector<TensorShape> ({armnnUtils::Permuted(inShape, m_Param.m_DimMappings)}); in InferOutputShapes()
DTransposeLayer.cpp42 …turn std::vector<TensorShape> ({armnnUtils::TransposeTensorShape(inShape, m_Param.m_DimMappings)}); in InferOutputShapes()
/external/armnn/src/backends/reference/workloads/
DRefTransposeWorkload.cpp39 const PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in Execute()
DRefPermuteWorkload.cpp39 const PermutationVector& mappings = m_Data.m_Parameters.m_DimMappings; in Execute()
/external/armnn/src/backends/tosaCommon/test/
DTosaTestUtils.hpp164 …std::vector<int> outPerm(transposeDesc->m_DimMappings.begin(), transposeDesc->m_DimMappings.end()); in VerifyTosaAttribute()
/external/armnn/src/armnn/optimizations/
DPermuteAndBatchToSpaceAsDepthToSpace.hpp33 …if (!static_cast<PermuteType&>(base).GetParameters().m_DimMappings.IsEqual(PermutationVector{ 3, 1… in Run()
/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_descriptors.i715 PermutationVector m_DimMappings; member
1226 PermutationVector m_DimMappings; member
/external/armnn/src/armnn/
DSerializeLayerParameters.cpp384 for (auto it : desc.m_DimMappings) in Serialize()
581 for (auto it : desc.m_DimMappings) in Serialize()
DDescriptors.cpp72 m_DimMappings[i] = dimMappings[i]; in PermutationVector()
/external/armnn/src/armnnSerializer/
DSerializer.cpp943 for (unsigned int i=0; i<permuteDescriptor.m_DimMappings.GetSize(); ++i) in SerializePermuteLayer()
945 dimMappings.push_back(permuteDescriptor.m_DimMappings[i]); in SerializePermuteLayer()
1479 for (unsigned int i=0; i<descriptor.m_DimMappings.GetSize(); ++i) in SerializeTransposeLayer()
1481 dimMappings.push_back(descriptor.m_DimMappings[i]); in SerializeTransposeLayer()
/external/armnn/shim/sl/canonical/
DConversionUtils.hpp569 transposeDesc.m_DimMappings = mapping; in TransposeInputTensors()

12