Home
last modified time | relevance | path

Searched refs:GetConstTensor (Results 1 – 24 of 24) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp121 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
123 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
125 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
129 …tToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
131 …rrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<void>()); in Execute()
133 …tToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
137 m_ForgetGateBiasTensor->GetTensorInfo(), m_ForgetGateBiasTensor->GetConstTensor<void>()); in Execute()
139 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetConstTensor<void>()); in Execute()
141 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
163 … m_InputLayerNormWeights->GetTensorInfo(), m_InputLayerNormWeights->GetConstTensor<void>()); in Execute()
[all …]
DRefUnidirectionalSequenceLstmWorkload.cpp152 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
154 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
156 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
160 …tToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
162 …rrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<void>()); in Execute()
164 …tToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
168 m_ForgetGateBiasTensor->GetTensorInfo(), m_ForgetGateBiasTensor->GetConstTensor<void>()); in Execute()
170 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetConstTensor<void>()); in Execute()
172 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
191 … m_InputLayerNormWeights->GetTensorInfo(), m_InputLayerNormWeights->GetConstTensor<void>()); in Execute()
[all …]
DRefQLstmWorkload.cpp116 …InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
118 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
120 …InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
124 m_RecurrentToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
126 …rrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<void>()); in Execute()
129 m_RecurrentToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
215 …m_InputToInputWeightsTensor->GetTensorInfo(), m_InputToInputWeightsTensor->GetConstTensor<void>()); in Execute()
217 … m_RecurrentToInputWeightsTensor->GetConstTensor<void>()); in Execute()
225 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetConstTensor<void>()); in Execute()
228 …m_CellToForgetWeightsTensor->GetTensorInfo(), m_CellToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
[all …]
DRefConstantWorkload.cpp36 …memcpy(outputs[0]->Map(), m_Data.m_LayerOutput->GetConstTensor<void>(), GetTensorInfo(outputs[0]).… in Execute()
/external/armnn/src/backends/neon/workloads/
DNeonConstantWorkload.cpp78 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<BFloat16>(), output); in Execute()
83 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<Half>(), output); in Execute()
88 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<float>(), output); in Execute()
93 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<uint8_t>(), output); in Execute()
98 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<int8_t>(), output); in Execute()
103 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<int16_t>(), output); in Execute()
109 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<int8_t>(), output); in Execute()
114 CopyArmComputeITensorData(data.m_LayerOutput->GetConstTensor<int32_t>(), output); in Execute()
DNeonWorkloadUtils.hpp104 CopyArmComputeTensorData(tensor, handle->GetConstTensor<armnn::Half>()); in InitializeArmComputeTensorData()
107 CopyArmComputeTensorData(tensor, handle->GetConstTensor<float>()); in InitializeArmComputeTensorData()
110 CopyArmComputeTensorData(tensor, handle->GetConstTensor<uint8_t>()); in InitializeArmComputeTensorData()
114 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int8_t>()); in InitializeArmComputeTensorData()
117 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int32_t>()); in InitializeArmComputeTensorData()
120 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int16_t>()); in InitializeArmComputeTensorData()
123 CopyArmComputeTensorData(tensor, handle->GetConstTensor<armnn::BFloat16>()); in InitializeArmComputeTensorData()
/external/armnn/src/backends/cl/workloads/
DClConstantWorkload.cpp72 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<Half>()); in Execute()
77 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<float>()); in Execute()
82 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<uint8_t>()); in Execute()
87 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int8_t>()); in Execute()
92 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int16_t>()); in Execute()
98 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int8_t>()); in Execute()
103 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int32_t>()); in Execute()
DClWorkloadUtils.hpp125 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<armnn::Half>()); in InitializeArmComputeClTensorData()
128 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<float>()); in InitializeArmComputeClTensorData()
131 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>()); in InitializeArmComputeClTensorData()
135 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int8_t>()); in InitializeArmComputeClTensorData()
138 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int16_t>()); in InitializeArmComputeClTensorData()
141 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>()); in InitializeArmComputeClTensorData()
144 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<armnn::BFloat16>()); in InitializeArmComputeClTensorData()
/external/armnn/include/armnn/backends/
DTensorHandle.hpp28 const T* GetConstTensor() const in GetConstTensor() function in armnn::ConstTensorHandle
76 const void* ConstTensorHandle::GetConstTensor<void>() const;
182 const void* ConstTensorHandle::GetConstTensor() const;
/external/armnn/src/armnn/test/optimizations/
DConvertConstantsFloatToHalfTests.cpp59 const Half* data = weightsLayer->m_LayerOutput->GetConstTensor<Half>();
119 const Half* data = weights->m_LayerOutput->GetConstTensor<Half>();
126 const Half* biasData = biases->m_LayerOutput->GetConstTensor<Half>();
DConvertConstantsHalfToFloatTests.cpp59 const float* data = weightsLayer->m_LayerOutput->GetConstTensor<float>();
/external/armnn/src/armnn/optimizations/
DConvertConstants.hpp31 armnnUtils::FloatingPointConverter::ConvertFloat16To32(handle->GetConstTensor<Half>(), in Func()
52 armnnUtils::FloatingPointConverter::ConvertFloat32To16(handle->GetConstTensor<float>(), in Func()
DAddBroadcastReshapeLayer.hpp75 … ConstTensor(reshapeInfo, constantLayer.m_LayerOutput.get()->GetConstTensor<void>())); in Run()
/external/armnn/src/backends/backendsCommon/
DTensorHandle.cpp38 const void* ConstTensorHandle::GetConstTensor<void>() const in GetConstTensor() function in armnn::ConstTensorHandle
69 CopyFrom(tensorHandle.GetConstTensor<void>(), tensorHandle.GetTensorInfo().GetNumBytes()); in ScopedTensorHandle()
DWorkloadUtils.cpp30 tensor->GetConstTensor<void>(), permuteBuffer, in PermuteTensor()
35 ::memcpy(permuteBuffer, tensor->GetConstTensor<void>(), tensorInfo.GetNumBytes()); in PermuteTensor()
/external/armnn/shim/sl/canonical/
DConversionUtils.cpp115 const armnn::ConstTensor& ConstTensorPin::GetConstTensor() const in GetConstTensor() function in armnn_driver::ConstTensorPin
302 tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
309 data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); in ConvertToLayerInputHandle()
312 armnn::TensorInfo constantTensorInfo = tensorPin.GetConstTensor().GetInfo(); in ConvertToLayerInputHandle()
DConversionUtils.hpp110 const armnn::ConstTensor& GetConstTensor() const;
DConverter.cpp2179 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertGroupedConv2d()
2180 ConstTensor biases = biasesPin.GetConstTensor(); in ConvertGroupedConv2d()
5486 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertTransposeConv2d()
5487 ConstTensor bias = biasPin.GetConstTensor(); in ConvertTransposeConv2d()
/external/armnn/src/armnn/test/
DGraphTests.cpp608 sharedWeightPtr = constantLayer->m_LayerOutput->GetConstTensor<float>();
629 const void* weightData = constLayer->GetConstantTensorsByRef()[0].get()->GetConstTensor<void>();
638 …void* resultDataWeights = weightsLayer->GetConstantTensorsByRef()[0].get()->GetConstTensor<void>();
DSubgraphViewTests.cpp1911 ->m_LayerOutput->GetConstTensor<void>(); in ReplaceConstantMultiplicationWithDepthwise()
/external/android-nn-driver/
DConversionUtils.cpp118 const armnn::ConstTensor& ConstTensorPin::GetConstTensor() const in GetConstTensor() function in armnn_driver::ConstTensorPin
DConversionUtils.hpp109 const armnn::ConstTensor& GetConstTensor() const;
1285 tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
1292 data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); in ConvertToLayerInputHandle()
1295 armnn::TensorInfo constantTensorInfo = tensorPin.GetConstTensor().GetInfo(); in ConvertToLayerInputHandle()
1419 tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
1426 data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); in ConvertToLayerInputHandle()
1429 armnn::TensorInfo constantTensorInfo = tensorPin.GetConstTensor().GetInfo(); in ConvertToLayerInputHandle()
DConversionUtils_1_2.hpp1030 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertGroupedConv2d()
1031 ConstTensor biases = biasesPin.GetConstTensor(); in ConvertGroupedConv2d()
2963 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertTransposeConv2d()
2964 ConstTensor bias = biasPin.GetConstTensor(); in ConvertTransposeConv2d()
/external/armnn/src/armnn/
DNetwork.cpp857 constantLayer->m_LayerOutput->GetConstTensor<Half>(), in AttemptBackendAssignment()