Home
last modified time | relevance | path

Searched refs:ScopedTensorHandle (Results 1 – 25 of 59) sorted by relevance

123

/external/armnn/src/backends/reference/workloads/
DRefQLstmWorkload.hpp26 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor;
27 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor;
28 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor;
29 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor;
31 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor;
32 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor;
33 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor;
34 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor;
36 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor;
37 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor;
[all …]
DRefUnidirectionalSequenceLstmWorkload.hpp31 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor;
32 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor;
33 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor;
34 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor;
35 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor;
36 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor;
37 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor;
38 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor;
39 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor;
40 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor;
[all …]
DRefLstmWorkload.hpp26 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor;
27 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor;
28 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor;
29 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor;
30 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor;
31 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor;
32 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor;
33 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor;
34 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor;
35 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor;
[all …]
DRefBatchNormalizationWorkload.hpp24 std::unique_ptr<ScopedTensorHandle> m_Mean;
25 std::unique_ptr<ScopedTensorHandle> m_Variance;
26 std::unique_ptr<ScopedTensorHandle> m_Beta;
27 std::unique_ptr<ScopedTensorHandle> m_Gamma;
DRefBatchNormalizationWorkload.cpp19 , m_Mean (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Mean))) in RefBatchNormalizationWorkload()
20 , m_Variance(std::make_unique<ScopedTensorHandle>(*(descriptor.m_Variance))) in RefBatchNormalizationWorkload()
21 , m_Beta (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Beta))) in RefBatchNormalizationWorkload()
22 , m_Gamma (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Gamma))) in RefBatchNormalizationWorkload()
DRefTransposeConvolution2dWorkload.hpp29 std::unique_ptr<ScopedTensorHandle> m_Weights;
30 std::unique_ptr<ScopedTensorHandle> m_Biases;
DRefTransposeConvolution2dWorkload.cpp21 m_Weights = std::make_unique<ScopedTensorHandle>(*(descriptor.m_Weight)); in RefTransposeConvolution2dWorkload()
30 m_Biases = std::make_unique<ScopedTensorHandle>(*(descriptor.m_Bias)); in RefTransposeConvolution2dWorkload()
/external/armnn/src/backends/backendsCommon/
DTensorHandle.cpp55 ScopedTensorHandle::ScopedTensorHandle(const TensorInfo& tensorInfo) in ScopedTensorHandle() function in armnn::ScopedTensorHandle
60 ScopedTensorHandle::ScopedTensorHandle(const ConstTensor& tensor) in ScopedTensorHandle() function in armnn::ScopedTensorHandle
61 : ScopedTensorHandle(tensor.GetInfo()) in ScopedTensorHandle()
66 ScopedTensorHandle::ScopedTensorHandle(const ConstTensorHandle& tensorHandle) in ScopedTensorHandle() function in armnn::ScopedTensorHandle
67 : ScopedTensorHandle(tensorHandle.GetTensorInfo()) in ScopedTensorHandle()
72 ScopedTensorHandle::ScopedTensorHandle(const ScopedTensorHandle& other) in ScopedTensorHandle() function in armnn::ScopedTensorHandle
78 ScopedTensorHandle& ScopedTensorHandle::operator=(const ScopedTensorHandle& other) in operator =()
86 ScopedTensorHandle::~ScopedTensorHandle() in ~ScopedTensorHandle()
91 void ScopedTensorHandle::Allocate() in Allocate()
104 void ScopedTensorHandle::CopyOutTo(void* memory) const in CopyOutTo()
[all …]
/external/armnn/src/backends/backendsCommon/test/layerTests/
DUnidirectionalSequenceLstmTestImpl.cpp140 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
141 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
142 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
143 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
144 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
145 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
146 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
147 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
148 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
149 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in UnidirectionalSequenceLstmTimeMajorSingleBatchTestImpl()
[all …]
DLstmTestImpl.cpp270 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
271 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
272 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
273 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo8); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
274 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
275 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
276 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
277 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo16); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
278 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
279 armnn::ScopedTensorHandle inputGateBiasTensor(tensorInfo4); in LstmNoCifgNoPeepholeNoProjectionTestImpl()
[all …]
DBatchNormalizationTestImpl.cpp75 armnn::ScopedTensorHandle meanTensor(tensorInfo); in BatchNormTestImpl()
76 armnn::ScopedTensorHandle varianceTensor(tensorInfo); in BatchNormTestImpl()
77 armnn::ScopedTensorHandle betaTensor(tensorInfo); in BatchNormTestImpl()
78 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in BatchNormTestImpl()
164 armnn::ScopedTensorHandle meanTensor(tensorInfo); in BatchNormTestNhwcImpl()
165 armnn::ScopedTensorHandle varianceTensor(tensorInfo); in BatchNormTestNhwcImpl()
166 armnn::ScopedTensorHandle betaTensor(tensorInfo); in BatchNormTestNhwcImpl()
167 armnn::ScopedTensorHandle gammaTensor(tensorInfo); in BatchNormTestNhwcImpl()
654 armnn::ScopedTensorHandle meanTensor(tensorInfo); in CompareBatchNormTest()
655 armnn::ScopedTensorHandle varianceTensor(tensorInfo); in CompareBatchNormTest()
[all …]
/external/armnn/src/armnn/test/
DShapeInferenceTests.cpp264 layer->m_LayerOutput = std::make_unique<ScopedTensorHandle>(output0);
376 layer->m_Anchors = std::make_unique<ScopedTensorHandle>(anchorsTensor);
458 … layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle>(constTensor);
459 …layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<ScopedTensorHandle>(constTensor…
460 layer->m_BasicParameters.m_CellBias = std::make_unique<ScopedTensorHandle>(constTensor);
461 layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<ScopedTensorHandle>(constTensor);
462 layer->m_CifgParameters.m_InputGateBias = std::make_unique<ScopedTensorHandle>(constTensor);
463 layer->m_BasicParameters.m_OutputGateBias = std::make_unique<ScopedTensorHandle>(constTensor);
464 …layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle>(constTe…
465 …layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<ScopedTensorHandle>(constTens…
[all …]
DOptimizerTests.cpp48 layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
50 layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
52 layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
54 layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
56 layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
58 layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
60 layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
62 layer->m_BasicParameters.m_CellBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
64 layer->m_BasicParameters.m_OutputGateBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
79 layer->m_CifgParameters.m_InputToInputWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()
[all …]
/external/armnn/src/backends/backendsCommon/test/
DIsLayerSupportedTestImpl.hpp88 m_Layer->m_Mean = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
90 m_Layer->m_Variance = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
92 m_Layer->m_Beta = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
94 m_Layer->m_Gamma = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
277 m_Layer->m_Weight = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
279 m_Layer->m_Bias = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
297 m_Layer->m_Anchors = std::make_unique<armnn::ScopedTensorHandle>( in DummyLayer()
318 …_Layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<armnn::ScopedTensorHandle>( in DummyLstmLayer()
320 …_Layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<armnn::ScopedTensorHandle>( in DummyLstmLayer()
322 …_Layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<armnn::ScopedTensorHandle>( in DummyLstmLayer()
[all …]
DDefaultAsyncExecuteTest.cpp137 ScopedTensorHandle syncInput0(constInputTensor);
138 ScopedTensorHandle syncOutput0(constOutputTensor);
144 ScopedTensorHandle asyncInput0(constInputTensor);
145 ScopedTensorHandle asyncOutput0(constOutputTensor);
178 ScopedTensorHandle defaultInput = ScopedTensorHandle(defaultTensor);
179 ScopedTensorHandle defaultOutput = ScopedTensorHandle(defaultTensor);
183 ScopedTensorHandle asyncInput(constInputTensor);
184 ScopedTensorHandle asyncOutput(constOutputTensor);
224 ScopedTensorHandle defaultInput = ScopedTensorHandle(defaultTensor);
225 ScopedTensorHandle defaultOutput = ScopedTensorHandle(defaultTensor);
[all …]
DLayerReleaseConstantDataTest.cpp36 layer->m_Mean = std::make_unique<ScopedTensorHandle>(weightInfo);
37 layer->m_Variance = std::make_unique<ScopedTensorHandle>(weightInfo);
38 layer->m_Beta = std::make_unique<ScopedTensorHandle>(weightInfo);
39 layer->m_Gamma = std::make_unique<ScopedTensorHandle>(weightInfo);
92 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
93 biasLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(biasInfo);
141 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
142 biasLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(biasInfo);
188 weightsLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(weightsInfo);
189 biasLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(biasInfo);
DWorkloadDataValidation.cpp35 ScopedTensorHandle sameTensor(sameInfo);
533 armnn::ScopedTensorHandle inputToInputWeightsTensor(tensorInfo4x5);
534 armnn::ScopedTensorHandle inputToForgetWeightsTensor(tensorInfo4x5);
535 armnn::ScopedTensorHandle inputToCellWeightsTensor(tensorInfo4x5);
536 armnn::ScopedTensorHandle inputToOutputWeightsTensor(tensorInfo4x5);
537 armnn::ScopedTensorHandle recurrentToForgetWeightsTensor(tensorInfo4x3);
538 armnn::ScopedTensorHandle recurrentToInputWeightsTensor(tensorInfo4x3);
539 armnn::ScopedTensorHandle recurrentToCellWeightsTensor(tensorInfo4x3);
540 armnn::ScopedTensorHandle recurrentToOutputWeightsTensor(tensorInfo4x3);
541 armnn::ScopedTensorHandle cellToInputWeightsTensor(tensorInfo4);
[all …]
/external/armnn/include/armnn/backends/
DTensorHandle.hpp115 class ScopedTensorHandle : public TensorHandle class
118 explicit ScopedTensorHandle(const TensorInfo& tensorInfo);
121 explicit ScopedTensorHandle(const ConstTensor& tensor);
124 explicit ScopedTensorHandle(const ConstTensorHandle& tensorHandle);
126 ScopedTensorHandle(const ScopedTensorHandle& other);
127 ScopedTensorHandle& operator=(const ScopedTensorHandle& other);
128 ~ScopedTensorHandle();
137 void CopyFrom(const ScopedTensorHandle& other);
/external/armnn/src/armnn/
DNetwork.cpp864 constantLayer->m_LayerOutput.reset(new ScopedTensorHandle(newInput)); in AttemptBackendAssignment()
2154 layer->m_Anchors = std::make_shared<ScopedTensorHandle>(anchors); in AddDetectionPostProcessLayer()
2247 layer->m_Mean = std::make_shared<ScopedTensorHandle>(mean); in AddBatchNormalizationLayer()
2248 layer->m_Variance = std::make_shared<ScopedTensorHandle>(variance); in AddBatchNormalizationLayer()
2249 layer->m_Beta = std::make_shared<ScopedTensorHandle>(beta); in AddBatchNormalizationLayer()
2250 layer->m_Gamma = std::make_shared<ScopedTensorHandle>(gamma); in AddBatchNormalizationLayer()
2298 layer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(input); in AddConstantLayer()
2334 std::make_shared<ScopedTensorHandle>(*(params.m_InputToForgetWeights)); in AddLstmLayer()
2336 std::make_shared<ScopedTensorHandle>(*(params.m_InputToCellWeights)); in AddLstmLayer()
2338 std::make_shared<ScopedTensorHandle>(*(params.m_InputToOutputWeights)); in AddLstmLayer()
[all …]
/external/armnn/src/backends/cl/test/
DOpenClTimerTest.cpp81 ScopedTensorHandle meanTensor(tensorInfo);
82 ScopedTensorHandle varianceTensor(tensorInfo);
83 ScopedTensorHandle betaTensor(tensorInfo);
84 ScopedTensorHandle gammaTensor(tensorInfo);
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp391 layer->m_Mean = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWorkloadTest()
392 layer->m_Variance = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWorkloadTest()
393 layer->m_Beta = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWorkloadTest()
394 layer->m_Gamma = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWorkloadTest()
449 layer->m_Mean = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWithBlobWorkloadTest()
450 layer->m_Variance = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWithBlobWorkloadTest()
451 layer->m_Beta = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWithBlobWorkloadTest()
452 layer->m_Gamma = std::make_unique<ScopedTensorHandle>(weightInfo); in CreateBatchNormalizationWithBlobWorkloadTest()
541 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
624 weights->m_LayerOutput = std::make_unique<ScopedTensorHandle>(weightsTensorInfo);
[all …]
/external/armnn/src/armnn/test/optimizations/
DConvertConstantsFloatToHalfTests.cpp39 weightsLayer->m_LayerOutput = std::make_unique<armnn::ScopedTensorHandle>(weights);
93 weights->m_LayerOutput = std::make_unique<armnn::ScopedTensorHandle>(weightsTensor);
94 biases->m_LayerOutput = std::make_unique<armnn::ScopedTensorHandle>(biasesTensor);
DConvertConstDequantisationLayersToConstLayersTest.cpp62 constantLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(constTensor);
98 constantLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(constTensor);
168 constantLayer->m_LayerOutput = std::make_shared<ScopedTensorHandle>(constTensor);
/external/armnn/src/dynamic/sample/
DSampleDynamicWorkloadFactory.cpp48 return std::make_unique<armnn::ScopedTensorHandle>(tensorInfo); in CreateTensorHandle()
56 return std::make_unique<armnn::ScopedTensorHandle>(tensorInfo); in CreateTensorHandle()
/external/armnn/src/armnn/optimizations/
DConvertConstants.hpp37 handle.reset(new ScopedTensorHandle(newInput)); in Func()
58 handle.reset(new ScopedTensorHandle(newInput)); in Func()

123