Home
last modified time | relevance | path

Searched refs:ConstTensorHandle (Results 1 – 23 of 23) sorted by relevance

/external/armnn/include/armnn/backends/
DWorkloadData.hpp241 const ConstTensorHandle* m_Anchors;
316 const ConstTensorHandle* m_Mean;
317 const ConstTensorHandle* m_Variance;
318 const ConstTensorHandle* m_Beta;
319 const ConstTensorHandle* m_Gamma;
342 const ConstTensorHandle* m_Min;
343 const ConstTensorHandle* m_Max;
370 const ConstTensorHandle* m_LayerOutput;
422 const ConstTensorHandle* m_InputToInputWeights;
423 const ConstTensorHandle* m_InputToForgetWeights;
[all …]
DTensorHandle.hpp24 class ConstTensorHandle : public ITensorHandle class
59 ConstTensorHandle(const TensorInfo& tensorInfo);
68 ConstTensorHandle(const ConstTensorHandle& other) = delete;
69 ConstTensorHandle& operator=(const ConstTensorHandle& other) = delete;
76 const void* ConstTensorHandle::GetConstTensor<void>() const;
79 class TensorHandle : public ConstTensorHandle
124 explicit ScopedTensorHandle(const ConstTensorHandle& tensorHandle);
166 class ConstPassthroughTensorHandle : public ConstTensorHandle
170 : ConstTensorHandle(tensorInfo) in ConstPassthroughTensorHandle()
182 const void* ConstTensorHandle::GetConstTensor() const;
[all …]
DTensorHandleFwd.hpp11 class ConstTensorHandle;
/external/armnn/src/armnn/layers/
DLstmParameters.hpp17 std::shared_ptr<ConstTensorHandle> m_InputLayerNormWeights;
19 std::shared_ptr<ConstTensorHandle> m_ForgetLayerNormWeights;
21 std::shared_ptr<ConstTensorHandle> m_CellLayerNormWeights;
23 std::shared_ptr<ConstTensorHandle> m_OutputLayerNormWeights;
29 std::shared_ptr<ConstTensorHandle> m_InputToInputWeights;
31 std::shared_ptr<ConstTensorHandle> m_RecurrentToInputWeights;
33 std::shared_ptr<ConstTensorHandle> m_InputGateBias;
39 std::shared_ptr<ConstTensorHandle> m_ProjectionWeights;
41 std::shared_ptr<ConstTensorHandle> m_ProjectionBias;
47 std::shared_ptr<ConstTensorHandle> m_CellToInputWeights;
[all …]
DQLstmLayer.hpp17 std::shared_ptr<ConstTensorHandle> m_InputToForgetWeights;
19 std::shared_ptr<ConstTensorHandle> m_InputToCellWeights;
21 std::shared_ptr<ConstTensorHandle> m_InputToOutputWeights;
24 std::shared_ptr<ConstTensorHandle> m_RecurrentToForgetWeights;
26 std::shared_ptr<ConstTensorHandle> m_RecurrentToCellWeights;
28 std::shared_ptr<ConstTensorHandle> m_RecurrentToOutputWeights;
31 std::shared_ptr<ConstTensorHandle> m_ForgetGateBias;
33 std::shared_ptr<ConstTensorHandle> m_CellBias;
35 std::shared_ptr<ConstTensorHandle> m_OutputGateBias;
41 std::shared_ptr<ConstTensorHandle> m_ProjectionWeights;
[all …]
DQuantizedLstmLayer.hpp17 std::shared_ptr<ConstTensorHandle> m_InputToInputWeights;
19 std::shared_ptr<ConstTensorHandle> m_InputToForgetWeights;
21 std::shared_ptr<ConstTensorHandle> m_InputToCellWeights;
23 std::shared_ptr<ConstTensorHandle> m_InputToOutputWeights;
26 std::shared_ptr<ConstTensorHandle> m_RecurrentToInputWeights;
28 std::shared_ptr<ConstTensorHandle> m_RecurrentToForgetWeights;
30 std::shared_ptr<ConstTensorHandle> m_RecurrentToCellWeights;
32 std::shared_ptr<ConstTensorHandle> m_RecurrentToOutputWeights;
35 std::shared_ptr<ConstTensorHandle> m_InputGateBias;
37 std::shared_ptr<ConstTensorHandle> m_ForgetGateBias;
[all …]
DBatchNormalizationLayer.hpp19 std::shared_ptr<ConstTensorHandle> m_Mean;
21 std::shared_ptr<ConstTensorHandle> m_Variance;
23 std::shared_ptr<ConstTensorHandle> m_Beta;
25 std::shared_ptr<ConstTensorHandle> m_Gamma;
DTransposeConvolution2dLayer.hpp19 std::shared_ptr<ConstTensorHandle> m_Weight;
21 std::shared_ptr<ConstTensorHandle> m_Bias;
DDetectionPostProcessLayer.hpp20 std::shared_ptr<ConstTensorHandle> m_Anchors;
DConstantLayer.hpp44 std::shared_ptr<ConstTensorHandle> m_LayerOutput;
/external/armnn/src/backends/backendsCommon/
DTensorHandle.cpp31 ConstTensorHandle::ConstTensorHandle(const TensorInfo& tensorInfo) in ConstTensorHandle() function in armnn::ConstTensorHandle
38 const void* ConstTensorHandle::GetConstTensor<void>() const in GetConstTensor()
44 : ConstTensorHandle(tensorInfo) in TensorHandle()
66 ScopedTensorHandle::ScopedTensorHandle(const ConstTensorHandle& tensorHandle) in ScopedTensorHandle()
DWorkloadUtils.hpp208 armnn::ConstTensor PermuteTensor(const ConstTensorHandle* tensor,
224 armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstTensorHandle* weightTensor,
237 std::tuple<ConstTensor, unsigned int> Convert1HWOTensorToAcl(const ConstTensorHandle* weightTensor,
249 std::tuple<ConstTensor, unsigned int> Convert1HWOtoMIHW(const ConstTensorHandle* weightTensor,
DWorkloadUtils.cpp18 armnn::ConstTensor PermuteTensor(const ConstTensorHandle* tensor, in PermuteTensor()
139 std::tuple<ConstTensor, unsigned int> Convert1HWOTensorToAcl(const ConstTensorHandle* weightTensor, in Convert1HWOTensorToAcl()
201 std::tuple<ConstTensor, unsigned int> Convert1HWOtoMIHW(const ConstTensorHandle* weightTensor, in Convert1HWOtoMIHW()
230 armnn::ConstTensor ConvertWeightTensorFromArmnnToAcl(const ConstTensorHandle* weightTensor, in ConvertWeightTensorFromArmnnToAcl()
/external/armnn/src/armnn/test/
DTestLayerVisitor.hpp26 const ConstTensorHandle& actual);
33 const std::shared_ptr<ConstTensorHandle> actual);
DTestLayerVisitor.cpp52 void TestLayerVisitor::CheckConstTensors(const ConstTensor& expected, const ConstTensorHandle& actu… in CheckConstTensors()
74 const std::shared_ptr<ConstTensorHandle> actual) in CheckConstTensorPtrs()
/external/armnn/src/armnn/optimizations/
DConvertConstants.hpp23 static void Func(std::shared_ptr<ConstTensorHandle>& handle) in Func()
44 static void Func(std::shared_ptr<ConstTensorHandle>& handle) in Func()
/external/armnn/src/backends/reference/workloads/
DLstmUtils.hpp88 std::unique_ptr<armnn::ScopedTensorHandle> AssignScopedTensorHandle(const armnn::ConstTensorHandle
DLstmUtils.cpp299 std::unique_ptr<armnn::ScopedTensorHandle> AssignScopedTensorHandle(const armnn::ConstTensorHandle in AssignScopedTensorHandle()
/external/armnn/src/backends/cl/workloads/
DClWorkloadUtils.hpp117 const ConstTensorHandle* handle) in InitializeArmComputeClTensorData()
/external/armnn/src/backends/neon/workloads/
DNeonWorkloadUtils.hpp97 const ConstTensorHandle* handle) in InitializeArmComputeTensorData()
/external/armnn/src/armnn/
DLayer.cpp307 OperateOnConstantTensors([](std::shared_ptr<ConstTensorHandle>& handle) in ReleaseConstantData()
549 res.push_back(const_cast<std::shared_ptr<ConstTensorHandle>&>(i.get())); in GetConstantTensorsByRef()
/external/armnn/src/backends/tosaCommon/operatorMappings/
DTosaOperatorUtils.hpp280 …or<uint8_t> ConvertConstantTensorDataToBuffer(const std::shared_ptr<ConstTensorHandle>& tensorHand… in ConvertConstantTensorDataToBuffer()
/external/armnn/include/armnn/
DINetwork.hpp124 using ConstantTensors = std::vector<std::reference_wrapper<std::shared_ptr<ConstTensorHandle>>>;
129 …bleConstantTensors = std::vector<std::reference_wrapper<const std::shared_ptr<ConstTensorHandle>>>;