Home
last modified time | relevance | path

Searched refs:GetNumElements (Results 1 – 25 of 112) sorted by relevance

12345

/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
Darmnn_tensor.i64 ") GetNumElements;
65 unsigned int GetNumElements() const;
80 std::string elm = "NumElements: " + std::to_string($self->GetNumElements()); in __str__()
153 ") GetNumElements;
154 unsigned int GetNumElements() const;
258 + ", NumElements: " + std::to_string($self->GetNumElements()) + "}"; in __str__()
281 unsigned int GetNumElements() const;
293 + ", NumElements: " + std::to_string($self->GetNumElements()) + "}"; in __str__()
316 unsigned int GetNumElements() const;
328 + ", NumElements: " + std::to_string($self->GetNumElements()) + "}"; in __str__()
/external/armnn/src/armnn/optimizations/
DConvertConstants.hpp32 std::vector<float> newValues(info.GetNumElements()); in Func()
35 info.GetNumElements(), in Func()
53 std::vector<float> newValues(info.GetNumElements()); in Func()
56 info.GetNumElements(), in Func()
74 std::vector<BFloat16> newValues(info.GetNumElements()); in Func()
77 info.GetNumElements(), in Func()
95 std::vector<Half> newValues(info.GetNumElements()); in Func()
98 info.GetNumElements(), in Func()
DFuseBatchNorm.hpp74 … std::vector<T> weightsVector (weightsBuffer, weightsBuffer + weightsTensor.GetNumElements()); in Run()
75 std::vector<T> betaVector (betaBuffer, betaBuffer + betaTensor.GetNumElements()); in Run()
76 std::vector<T> gammaVector (gammaBuffer, gammaBuffer + gammaTensor.GetNumElements()); in Run()
77 std::vector<T> meanVector (meanBuffer, meanBuffer + meanTensor.GetNumElements()); in Run()
78 std::vector<T> varianceVector(varBuffer, varBuffer + varTensor.GetNumElements()); in Run()
139 std::vector<T> biasVector(biasBuffer, biasBuffer + biasTensor.GetNumElements()); in Run()
/external/armnn/python/pyarmnn/test/
Dtest_tensor.py21 elements = tensor_info.GetNumElements()
28 assert elements == tensor.GetNumElements()
58 elements = tensor_info.GetNumElements()
72 assert 0 == tensor.GetNumElements()
88 assert copied_tensor.GetNumElements() == tensor.GetNumElements()
103 assert copied_tensor.GetNumElements() == tensor.GetNumElements()
142 assert tensor.GetNumElements() == 6
Dtest_tensor_conversion.py47 …input_data.append(np.random.randint(0, 255, size=(1, tensor_info.GetNumElements())).astype(np.uint…
78 assert data[i].size == output_tensors[i][1].GetNumElements()
87 …input_data.append(np.random.randint(0, 255, size=(1, tensor_info.GetNumElements())).astype(np.floa…
98 assert tensor[1].GetNumElements() == 28*28*1
Dtest_tensor_shape.py11 assert 6 == tensor_shape.GetNumElements()
17 assert 10 == tensor_shape.GetNumElements()
31 assert 6 == tensor_shape.GetNumElements()
Dtest_const_tensor.py68 assert data.size == tensor.GetNumElements()
82 assert copied_tensor.GetNumElements() == tensor.GetNumElements()
173 assert copied_tensor.GetNumElements() == tensor.GetNumElements()
182 num_elements = tensor_info.GetNumElements()
190 elements = tensor_info.GetNumElements()
197 assert elements == tensor.GetNumElements()
Dtest_tensor_info.py11 assert 2 == tensor_info.GetNumElements()
19 assert 2 == shape.GetNumElements()
/external/armnn/src/armnn/test/
DTensorTest.cpp266 …BOOST_CHECK_MESSAGE(1 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements(… in BOOST_AUTO_TEST_CASE()
293 BOOST_CHECK_THROW( shape.GetNumElements(), InvalidArgumentException ); in BOOST_AUTO_TEST_CASE()
312 …BOOST_CHECK_MESSAGE(0 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements(… in BOOST_AUTO_TEST_CASE()
337 …BOOST_CHECK_MESSAGE(6 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements(… in BOOST_AUTO_TEST_CASE()
388 …BOOST_CHECK_MESSAGE(6 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements(… in BOOST_AUTO_TEST_CASE()
418 …BOOST_CHECK_MESSAGE(30 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements… in BOOST_AUTO_TEST_CASE()
430 …BOOST_CHECK_MESSAGE( 0 == shape.GetNumElements(), "Number of elements is " << shape.GetNumElements… in BOOST_AUTO_TEST_CASE()
437 …BOOST_CHECK_MESSAGE( 0 == tensor.GetNumElements(), "Number of elements is " << tensor.GetNumElemen… in BOOST_AUTO_TEST_CASE()
440 BOOST_CHECK_MESSAGE( 0 == tensor.GetShape().GetNumElements(), "Number of dimensions is " << in BOOST_AUTO_TEST_CASE()
441 tensor.GetShape().GetNumElements()); in BOOST_AUTO_TEST_CASE()
DTestLayerVisitor.cpp37 BOOST_CHECK(expected.GetNumElements() == actual.GetNumElements()); in CheckConstTensors()
/external/armnn/src/backends/reference/workloads/
DDequantize.cpp19 ARMNN_ASSERT(inputInfo.GetNumElements() == outputInfo.GetNumElements()); in Dequantize()
20 for (unsigned int i = 0; i < inputInfo.GetNumElements(); i++) in Dequantize()
DBaseIterator.hpp143 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
181 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
219 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
257 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
297 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
336 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
372 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
401 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
438 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
471 const unsigned int size = tensorShape.GetNumElements(); in DecodeTensor()
[all …]
DRefWorkloadUtils.hpp91 std::vector<float> ret(info.GetNumElements()); in Dequantize()
92 for (size_t i = 0; i < info.GetNumElements(); i++) in Dequantize()
102 for (unsigned int i = 0; i < info.GetNumElements(); i++) in Dequantize()
110 for (size_t i = 0; i < info.GetNumElements(); i++) in Quantize()
DGather.cpp37 for (unsigned int i = 0; i < indicesInfo.GetNumElements(); ++i) in Gather()
56 ARMNN_ASSERT(outIndex == outputInfo.GetNumElements()); in Gather()
/external/android-nn-driver/test/1.1/
DTranspose.cpp64 inloc.length = inputs.GetNumElements() * sizeof(float); in TransposeTestImpl()
73 outloc.length = expectedOutputTensor.GetNumElements() * sizeof(float); in TransposeTestImpl()
84 AddPoolAndSetData(inputs.GetNumElements(), in TransposeTestImpl()
89 …android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutputTensor.GetNumElements(), r… in TransposeTestImpl()
98 for (unsigned int i = 0; i < expectedOutputTensor.GetNumElements(); ++i) in TransposeTestImpl()
DMean.cpp67 inLoc.length = input.GetNumElements() * sizeof(float); in MeanTestImpl()
76 outLoc.length = expectedOutput.GetNumElements() * sizeof(float); in MeanTestImpl()
87 AddPoolAndSetData(input.GetNumElements(), request, input.GetData()); in MeanTestImpl()
90 …android::sp<IMemory> outMemory = AddPoolAndGetData<float>(expectedOutput.GetNumElements(), request… in MeanTestImpl()
100 for (unsigned int i = 0; i < expectedOutput.GetNumElements(); i++) in MeanTestImpl()
/external/armnn/src/armnnSerializer/test/
DSerializerTests.cpp210 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
215 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
219 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
223 tensor1.GetMemoryArea(), tensor2.GetMemoryArea(), tensor1.GetNumElements()); in CompareConstTensor()
514 std::vector<float> constantData = GenerateRandomData<float>(info.GetNumElements()); in BOOST_AUTO_TEST_CASE()
588 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
591 std::vector<float> biasesData = GenerateRandomData<float>(biasesInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
685 std::vector<int8_t> kernelData = GenerateRandomData<int8_t>(kernelInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
687 std::vector<int32_t> biasData = GenerateRandomData<int32_t>(biasInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
803 std::vector<float> weightsData = GenerateRandomData<float>(weightsInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
[all …]
/external/armnn/include/armnn/
DTensor.hpp87 unsigned int GetNumElements() const;
192 unsigned int GetNumElements() const { return m_Shape.GetNumElements(); } in GetNumElements() function in armnn::TensorInfo
290 unsigned int GetNumElements() const { return m_Info.GetNumElements(); } in GetNumElements() function in armnn::BaseTensor
/external/armnn/src/armnnTfLiteParser/test/
DOutputShapeOfSqueeze.cpp34 BOOST_TEST(outputTensorInfo.GetNumElements() == 4); in BOOST_FIXTURE_TEST_CASE()
45 BOOST_TEST(outputTensorInfo.GetNumElements() == 4); in BOOST_FIXTURE_TEST_CASE()
56 BOOST_TEST(outputTensorInfo.GetNumElements() == 4); in BOOST_FIXTURE_TEST_CASE()
/external/armnn/src/armnnUtils/
DTensorIOUtils.hpp38 if (value.size() != inputBinding.second.GetNumElements()) in MakeInputTensors()
41 inputBinding.second.GetNumElements(), in MakeInputTensors()
76 if (value.size() != outputBinding.second.GetNumElements()) in MakeOutputTensors()
/external/android-nn-driver/test/
DTestTensor.hpp22 ARMNN_ASSERT(m_Shape.GetNumElements() == m_Data.size());
26 unsigned int GetNumElements() const;
DTestTensor.cpp21 unsigned int TestTensor::GetNumElements() const in GetNumElements() function in driverTestHelpers::TestTensor
23 return m_Shape.GetNumElements(); in GetNumElements()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DDetectionPostProcessTestImpl.hpp237 for (size_t i = 0; i < info.GetNumElements(); i++) in QuantizeData()
279 std::vector<RawType> boxEncodingsData(TestData::s_BoxEncodingsShape.GetNumElements()); in DetectionPostProcessRegularNmsQuantizedTest()
284 std::vector<RawType> scoresData(TestData::s_ScoresShape.GetNumElements()); in DetectionPostProcessRegularNmsQuantizedTest()
289 std::vector<RawType> anchorsData(TestData::s_AnchorsShape.GetNumElements()); in DetectionPostProcessRegularNmsQuantizedTest()
343 std::vector<RawType> boxEncodingsData(TestData::s_BoxEncodingsShape.GetNumElements()); in DetectionPostProcessFastNmsQuantizedTest()
348 std::vector<RawType> scoresData(TestData::s_ScoresShape.GetNumElements()); in DetectionPostProcessFastNmsQuantizedTest()
353 std::vector<RawType> anchorsData(TestData::s_AnchorsShape.GetNumElements()); in DetectionPostProcessFastNmsQuantizedTest()
/external/armnn/src/backends/reference/test/
DArgMinMaxTests.cpp18 std::vector<int64_t> outputValues(outputInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
41 std::vector<int64_t> outputValues(outputInfo.GetNumElements()); in BOOST_AUTO_TEST_CASE()
DRefDetectionPostProcessTests.cpp131 std::vector<float> detectionBoxes(detectionBoxesInfo.GetNumElements()); in DetectionPostProcessTestImpl()
132 std::vector<float> detectionScores(detectionScoresInfo.GetNumElements()); in DetectionPostProcessTestImpl()
133 std::vector<float> detectionClasses(detectionClassesInfo.GetNumElements()); in DetectionPostProcessTestImpl()

12345