| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadFactory.cpp | 101 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
102 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
113 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
114 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
115 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
129 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
130 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
143 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
144 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
145 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
[all …]
|
| D | TensorHandle.cpp | 67 : ScopedTensorHandle(tensorHandle.GetTensorInfo()) in ScopedTensorHandle()
69 CopyFrom(tensorHandle.GetConstTensor<void>(), tensorHandle.GetTensorInfo().GetNumBytes()); in ScopedTensorHandle()
73 : TensorHandle(other.GetTensorInfo()) in ScopedTensorHandle()
95 SetMemory(::operator new(GetTensorInfo().GetNumBytes())); in Allocate()
106 memcpy(memory, GetTensor<void>(), GetTensorInfo().GetNumBytes()); in CopyOutTo()
111 memcpy(GetTensor<void>(), memory, GetTensorInfo().GetNumBytes()); in CopyInFrom()
116 CopyFrom(other.GetTensor<void>(), other.GetTensorInfo().GetNumBytes()); in CopyFrom()
122 ARMNN_ASSERT(GetTensorInfo().GetNumBytes() == numBytes); in CopyFrom()
|
| /external/armnn/src/backends/neon/ |
| D | NeonBackend.cpp | 197 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
201 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
202 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
204 … baseLayer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
229 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
233 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
234 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
236 … baseLayer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
260 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
264 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
[all …]
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefLstmWorkload.cpp | 58 const TensorInfo& inputInfo = GetTensorInfo(inputs[0]); in Execute()
59 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]); in Execute()
121 …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 …m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
131 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<vo… in Execute()
133 …m_RecurrentToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetConstTenso… in Execute()
137 m_ForgetGateBiasTensor->GetTensorInfo(), m_ForgetGateBiasTensor->GetConstTensor<void>()); in Execute()
139 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetConstTensor<void>()); in Execute()
[all …]
|
| D | RefUnidirectionalSequenceLstmWorkload.cpp | 60 TensorInfo inputInfo = GetTensorInfo(inputs[0]); in Execute()
61 const TensorInfo& outputStateInfo = GetTensorInfo(inputs[1]); in Execute()
62 const TensorInfo& cellStateInfo = GetTensorInfo(inputs[2]); in Execute()
63 TensorInfo outputStateOutInfo = GetTensorInfo(outputs[0]); in Execute()
64 TensorInfo cellStateOutInfo = GetTensorInfo(outputs[1]); in Execute()
65 TensorInfo outputInfo = GetTensorInfo(outputs[2]); in Execute()
152 …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 …m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
[all …]
|
| D | RefQLstmWorkload.cpp | 66 const TensorInfo& inputInfo = GetTensorInfo(inputs[0]); in Execute()
67 const TensorInfo& outputStateInInfo = GetTensorInfo(inputs[1]); in Execute()
68 const TensorInfo& cellStateInInfo = GetTensorInfo(inputs[2]); in Execute()
70 const TensorInfo& outputStateOutInfo = GetTensorInfo(outputs[0]); in Execute()
71 const TensorInfo& cellStateOutInfo = GetTensorInfo(outputs[1]); in Execute()
72 const TensorInfo& outputInfo = GetTensorInfo(outputs[2]); in Execute()
116 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
118 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
120 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
123 m_RecurrentToForgetWeightsTensor->GetTensorInfo(), in Execute()
[all …]
|
| D | RefDetectionPostProcessWorkload.cpp | 37 const TensorInfo& boxEncodingsInfo = GetTensorInfo(inputs[0]); in Execute()
38 const TensorInfo& scoresInfo = GetTensorInfo(inputs[1]); in Execute()
39 const TensorInfo& anchorsInfo = m_Anchors->GetTensorInfo(); in Execute()
41 const TensorInfo& detectionBoxesInfo = GetTensorInfo(outputs[0]); in Execute()
42 const TensorInfo& detectionClassesInfo = GetTensorInfo(outputs[1]); in Execute()
43 const TensorInfo& detectionScoresInfo = GetTensorInfo(outputs[2]); in Execute()
44 const TensorInfo& numDetectionsInfo = GetTensorInfo(outputs[3]); in Execute()
|
| D | RefBatchMatMulWorkload.cpp | 34 const TensorInfo& inputXInfo = GetTensorInfo(inputs[0]); in Execute()
35 const TensorInfo& inputYInfo = GetTensorInfo(inputs[1]); in Execute()
36 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]); in Execute()
38 std::unique_ptr<Decoder<float>> inputXDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), in Execute()
41 std::unique_ptr<Decoder<float>> inputYDecoder = MakeDecoder<float>(GetTensorInfo(inputs[1]), in Execute()
44 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), in Execute()
|
| D | RefPreluWorkload.cpp | 36 const TensorInfo& inputInfo = GetTensorInfo(inputs[0]); in Execute()
37 const TensorInfo& alphaInfo = GetTensorInfo(inputs[1]); in Execute()
38 const TensorInfo& outputInfo = GetTensorInfo(outputs[0]); in Execute()
40 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), in Execute()
42 std::unique_ptr<Decoder<float>> alphaDecoder = MakeDecoder<float>(GetTensorInfo(inputs[1]), in Execute()
44 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), in Execute()
|
| D | RefDepthwiseConvolution2dWorkload.cpp | 55 const TensorShape& inputShape = GetTensorInfo(inputs[0]).GetShape(); in Execute()
56 const TensorShape& outputShape = GetTensorInfo(outputs[0]).GetShape(); in Execute()
57 const TensorShape& filterShape = GetTensorInfo(inputs[1]).GetShape(); in Execute()
59 …std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), input… in Execute()
60 …std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), outp… in Execute()
61 …std::unique_ptr<Decoder<float>> filterDecoder = MakeDecoder<float>(GetTensorInfo(inputs[1]), input… in Execute()
65 biasDecoder = MakeDecoder<float>(GetTensorInfo(inputs[2]), inputs[2]->Map()); in Execute()
|
| D | RefConvolution3dWorkload.cpp | 50 …std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), inputs… in Execute()
51 …std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), outp… in Execute()
53 const TensorShape& inputShape = GetTensorInfo(inputs[0]).GetShape(); in Execute()
54 const TensorShape& outputShape = GetTensorInfo(outputs[0]).GetShape(); in Execute()
56 const auto& filterInfo = GetTensorInfo(inputs[1]); in Execute()
57 …std::unique_ptr<Decoder<float>> filterDecoder = MakeDecoder<float>(GetTensorInfo(inputs[1]), input… in Execute()
62 biasDecoder = MakeDecoder<float>(GetTensorInfo(inputs[2]), inputs[2]->Map()); in Execute()
|
| D | RefBatchNormalizationWorkload.cpp | 41 std::unique_ptr<Decoder<float>> meanDecoder = MakeDecoder<float>(m_Mean->GetTensorInfo(), in Execute()
43 … std::unique_ptr<Decoder<float>> varianceDecoder = MakeDecoder<float>(m_Variance->GetTensorInfo(), in Execute()
45 std::unique_ptr<Decoder<float>> gammaDecoder = MakeDecoder<float>(m_Gamma->GetTensorInfo(), in Execute()
47 std::unique_ptr<Decoder<float>> betaDecoder = MakeDecoder<float>(m_Beta->GetTensorInfo(), in Execute()
49 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]), in Execute()
51 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]), in Execute()
|
| /external/armnn/src/backends/cl/ |
| D | ClBackend.cpp | 345 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
349 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
350 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
352 … baseLayer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
377 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
381 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
382 … activationLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
384 … baseLayer->GetInputSlot(1).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
408 … biases = baseLayer->GetInputSlot(2).GetConnectedOutputSlot()->GetTensorInfo(); in OptimizeSubgraphView()
412 … baseLayer->GetInputSlot(0).GetConnectedOutputSlot()->GetTensorInfo(), in OptimizeSubgraphView()
[all …]
|
| /external/armnn/src/armnn/layers/ |
| D | LstmLayer.cpp | 172 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
177 GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
178 GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
179 GetInputSlot(2).GetConnection()->GetTensorInfo().GetShape() in ValidateTensorShapesFromInputs()
248 …GetOutputSlot(1).GetTensorInfo().GetShape(), inferredShapes[1], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
250 …GetOutputSlot(2).GetTensorInfo().GetShape(), inferredShapes[2], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
252 …GetOutputSlot(3).GetTensorInfo().GetShape(), inferredShapes[3], m_ShapeInferenceMethod, "LstmLayer… in ValidateTensorShapesFromInputs()
343 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
348 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
353 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | QLstmLayer.cpp | 174 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
180 GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), // input in ValidateTensorShapesFromInputs()
181 GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape(), // previousOutputIn in ValidateTensorShapesFromInputs()
182 GetInputSlot(2).GetConnection()->GetTensorInfo().GetShape() // previousCellStateIn in ValidateTensorShapesFromInputs()
252 …GetOutputSlot(1).GetTensorInfo().GetShape(), inferredShapes[1], m_ShapeInferenceMethod, "QLstmLaye… in ValidateTensorShapesFromInputs()
254 …GetOutputSlot(2).GetTensorInfo().GetShape(), inferredShapes[2], m_ShapeInferenceMethod, "QLstmLaye… in ValidateTensorShapesFromInputs()
343 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
348 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
353 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
359 managedRecurrentToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | QuantizedLstmLayer.cpp | 100 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
106 GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), // input in ValidateTensorShapesFromInputs()
107 GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape(), // previousCellStateIn in ValidateTensorShapesFromInputs()
108 GetInputSlot(2).GetConnection()->GetTensorInfo().GetShape() // previousOutputIn in ValidateTensorShapesFromInputs()
144 ValidateAndCopyShape(GetOutputSlot(1).GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
195 constTensors.emplace_back(ConstTensor(managedInputToInputWeights.GetTensorInfo(), in ExecuteStrategy()
201 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
207 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
213 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
221 managedRecurrentToInputWeights.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | UnidirectionalSequenceLstmLayer.cpp | 174 const TensorShape& outputShape = GetOutputSlot(2).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
179 GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
180 GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
181 GetInputSlot(2).GetConnection()->GetTensorInfo().GetShape() in ValidateTensorShapesFromInputs()
350 constTensors.emplace_back(ConstTensor(managedInputToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
355 constTensors.emplace_back(ConstTensor(managedInputToCellWeights.GetTensorInfo(), in ExecuteStrategy()
360 constTensors.emplace_back(ConstTensor(managedInputToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
366 managedRecurrentToForgetWeights.GetTensorInfo(), in ExecuteStrategy()
372 managedRecurrentToCellWeights.GetTensorInfo(), in ExecuteStrategy()
378 managedRecurrentToOutputWeights.GetTensorInfo(), in ExecuteStrategy()
[all …]
|
| D | DetectionPostProcessLayer.cpp | 43 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
53 { GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
54 GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape() }); in ValidateTensorShapesFromInputs()
64 ValidateAndCopyShape(GetOutputSlot(1).GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
69 ValidateAndCopyShape(GetOutputSlot(2).GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
74 ValidateAndCopyShape(GetOutputSlot(3).GetTensorInfo().GetShape(), in ValidateTensorShapesFromInputs()
101 …std::vector<armnn::ConstTensor> constTensors { {managedAnchors.GetTensorInfo(), managedAnchors.Map… in ExecuteStrategy()
|
| D | BatchNormalizationLayer.cpp | 56 const TensorShape& outputShape = GetOutputSlot(0).GetTensorInfo().GetShape(); in ValidateTensorShapesFromInputs()
60 …auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShap… in ValidateTensorShapesFromInputs()
81 … std::vector<armnn::ConstTensor> constTensors { { managedMean.GetTensorInfo(), managedMean.Map() }, in ExecuteStrategy()
82 … { managedVariance.GetTensorInfo(), managedVariance.Map() }, in ExecuteStrategy()
83 … { managedBeta.GetTensorInfo(), managedBeta.Map() }, in ExecuteStrategy()
84 … { managedGamma.GetTensorInfo(), managedGamma.Map() } }; in ExecuteStrategy()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClQLstmWorkload.cpp | 32 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in ClQLstmWorkload()
35 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQLstmWorkload()
38 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in ClQLstmWorkload()
41 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in ClQLstmWorkload()
44 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in ClQLstmWorkload()
47 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in ClQLstmWorkload()
50 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClQLstmWorkload()
53 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClQLstmWorkload()
56 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQLstmWorkload()
66 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClQLstmWorkload()
[all …]
|
| D | ClQuantizedLstmWorkload.cpp | 70 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
73 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
76 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
79 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
82 …ComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
85 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
88 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
91 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
94 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
97 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
[all …]
|
| D | ClLstmFloatWorkload.cpp | 38 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in ClLstmFloatWorkload()
41 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClLstmFloatWorkload()
44 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
47 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in ClLstmFloatWorkload()
50 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in ClLstmFloatWorkload()
53 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
56 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
59 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClLstmFloatWorkload()
62 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
68 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
[all …]
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonQLstmWorkload.cpp | 30 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in NeonQLstmWorkload()
33 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQLstmWorkload()
36 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in NeonQLstmWorkload()
39 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in NeonQLstmWorkload()
42 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in NeonQLstmWorkload()
45 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in NeonQLstmWorkload()
48 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonQLstmWorkload()
51 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonQLstmWorkload()
54 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonQLstmWorkload()
64 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonQLstmWorkload()
[all …]
|
| D | NeonQuantizedLstmWorkload.cpp | 23 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
26 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
29 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
32 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
35 …ComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
38 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
41 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
44 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
47 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
50 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
[all …]
|
| D | NeonLstmFloatWorkload.cpp | 33 …ildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
36 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
39 …ildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
42 …mputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
45 …rmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
48 …mputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
51 BuildArmComputeTensor(*m_ForgetGateBiasTensor, m_Data.m_ForgetGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
54 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonLstmFloatWorkload()
57 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
63 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
[all …]
|