Home
last modified time | relevance | path

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

/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.hpp43 CopyArmComputeTensorData(tensor, handle->GetConstTensor<armnn::Half>()); in InitializeArmComputeTensorData()
46 CopyArmComputeTensorData(tensor, handle->GetConstTensor<float>()); in InitializeArmComputeTensorData()
49 CopyArmComputeTensorData(tensor, handle->GetConstTensor<uint8_t>()); in InitializeArmComputeTensorData()
56 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int8_t>()); in InitializeArmComputeTensorData()
60 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int32_t>()); in InitializeArmComputeTensorData()
63 CopyArmComputeTensorData(tensor, handle->GetConstTensor<int16_t>()); in InitializeArmComputeTensorData()
/external/armnn/src/backends/cl/workloads/
DClConstantWorkload.cpp70 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<Half>()); in Execute()
75 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<float>()); in Execute()
80 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<uint8_t>()); in Execute()
85 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int8_t>()); in Execute()
90 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int16_t>()); in Execute()
96 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int8_t>()); in Execute()
101 CopyArmComputeClTensorData(output, data.m_LayerOutput->GetConstTensor<int32_t>()); in Execute()
DClWorkloadUtils.hpp99 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<armnn::Half>()); in InitializeArmComputeClTensorData()
102 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<float>()); in InitializeArmComputeClTensorData()
105 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<uint8_t>()); in InitializeArmComputeClTensorData()
112 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int8_t>()); in InitializeArmComputeClTensorData()
115 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int16_t>()); in InitializeArmComputeClTensorData()
119 CopyArmComputeClTensorData(clTensor, handle->GetConstTensor<int32_t>()); in InitializeArmComputeClTensorData()
/external/armnn/src/backends/backendsCommon/
DCpuTensorHandle.hpp31 const T* GetConstTensor() const in GetConstTensor() function in armnn::ConstCpuTensorHandle
73 const void* ConstCpuTensorHandle::GetConstTensor<void>() const;
173 const void* ConstCpuTensorHandle::GetConstTensor() const;
DCpuTensorHandle.cpp38 const void* ConstCpuTensorHandle::GetConstTensor<void>() const in GetConstTensor() function in armnn::ConstCpuTensorHandle
69 CopyFrom(tensorHandle.GetConstTensor<void>(), tensorHandle.GetTensorInfo().GetNumBytes()); in ScopedCpuTensorHandle()
DWorkloadUtils.cpp26 tensor->GetConstTensor<void>(), permuteBuffer, in PermuteTensor()
31 ::memcpy(permuteBuffer, tensor->GetConstTensor<void>(), tensorInfo.GetNumBytes()); in PermuteTensor()
/external/armnn/src/backends/reference/workloads/
DRefConstantWorkload.cpp32 memcpy(GetOutputTensorData<void>(0, data), data.m_LayerOutput->GetConstTensor<void>(), in PostAllocationConfigure()
/external/armnn/src/armnn/layers/
DDetectionPostProcessLayer.cpp83 ConstTensor anchorTensor(m_Anchors->GetTensorInfo(), m_Anchors->GetConstTensor<void>()); in Accept()
DFullyConnectedLayer.cpp97 ConstTensor biasTensor(m_Bias->GetTensorInfo(), m_Bias->GetConstTensor<void>()); in Accept()
/external/armnn/src/armnnTfParser/
DTfParser.cpp875 ConstTensor GetConstTensor(std::vector<T>& outputTensorData) const in GetConstTensor() function in armnnTfParser::ParsedConstTfOperation
1636 ConstTensor scaleTensor = scaleNode->GetConstTensor(scaleTensorData); in ParseFusedBatchNorm()
1639 ConstTensor offsetTensor = offsetNode->GetConstTensor(offsetTensorData); in ParseFusedBatchNorm()
1642 ConstTensor meanTensor = meanNode->GetConstTensor(meanTensorData); in ParseFusedBatchNorm()
1645 ConstTensor varianceTensor = varianceNode->GetConstTensor(varianceTensorData); in ParseFusedBatchNorm()
1696 ConstTensor const_tensor = alpha->GetConstTensor(const_data); in IsSupportedLeakyReluPattern()
2077 permuteVectorInput->GetConstTensor(permuteVectorData); in ParseTranspose()
2169 ConstTensor paddingTensor = paddingTensorOp->GetConstTensor(paddingTensorData); in ParsePad()
2234 shapeNode->GetConstTensor(axisTensorData); in ParseConcat()
2362 ConstTensor shapeTensor = shapeNode->GetConstTensor(shapeTensorData); in ParseReshape()
[all …]
/external/android-nn-driver/
DConversionUtils.cpp91 const armnn::ConstTensor& ConstTensorPin::GetConstTensor() const in GetConstTensor() function in armnn_driver::ConstTensorPin
DConversionUtils.hpp103 const armnn::ConstTensor& GetConstTensor() const;
1241 tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
1248 data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); in ConvertToLayerInputHandle()
1250 outputSlot.SetTensorInfo(tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
1368 tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
1375 data.m_Network->AddConstantLayer(tensorPin.GetConstTensor()); in ConvertToLayerInputHandle()
1377 outputSlot.SetTensorInfo(tensorPin.GetConstTensor().GetInfo()); in ConvertToLayerInputHandle()
2364 armnn::ConstTensor weights = weightsPin.GetConstTensor(); in ConvertConv2d()
2365 armnn::ConstTensor bias = biasPin.GetConstTensor(); in ConvertConv2d()
2591 armnn::ConstTensor weights = weightsPin.GetConstTensor(); in ConvertDepthwiseConv2d()
[all …]
DConversionUtils_1_2.hpp257 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertConv2d_1_2()
258 ConstTensor bias = biasPin.GetConstTensor(); in ConvertConv2d_1_2()
441 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertDepthwiseConv2d_1_2()
442 ConstTensor bias = biasPin.GetConstTensor(); in ConvertDepthwiseConv2d_1_2()
837 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertGroupedConv2d()
838 ConstTensor biases = biasesPin.GetConstTensor(); in ConvertGroupedConv2d()
2744 ConstTensor weights = weightsPin.GetConstTensor(); in ConvertTransposeConv2d()
2745 ConstTensor bias = biasPin.GetConstTensor(); in ConvertTransposeConv2d()