| /external/armnn/src/backends/cl/workloads/ |
| D | ClQLstmWorkload.cpp | 99 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
103 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in ClQLstmWorkload()
117 … m_Data.m_InputLayerNormWeights != nullptr ? m_InputLayerNormWeightsTensor.get() : nullptr, in ClQLstmWorkload()
229 … InitializeArmComputeClTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in ClQLstmWorkload()
421 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | ClLstmFloatWorkload.cpp | 116 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
123 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in ClLstmFloatWorkload()
130 m_InputLayerNormWeightsTensor.get(), in ClLstmFloatWorkload()
224 … InitializeArmComputeClTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in ClLstmFloatWorkload()
390 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | ClQLstmWorkload.hpp | 52 std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor; member in armnn::ClQLstmWorkload
|
| D | ClLstmFloatWorkload.hpp | 51 std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor; member in armnn::ClLstmFloatWorkload
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 232 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
235 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in ClUnidirectionalSequenceLstmFloatWorkload()
247 …rmWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeightsTensor.get(); in ClUnidirectionalSequenceLstmFloatWorkload()
391 … InitializeArmComputeClTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
896 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 67 std::unique_ptr<arm_compute::CLTensor> m_InputLayerNormWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonQLstmWorkload.cpp | 97 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
101 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in NeonQLstmWorkload()
115 … m_Data.m_InputLayerNormWeights != nullptr ? m_InputLayerNormWeightsTensor.get() : nullptr, in NeonQLstmWorkload()
223 … InitializeArmComputeTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in NeonQLstmWorkload()
417 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | NeonLstmFloatWorkload.cpp | 111 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
114 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in NeonLstmFloatWorkload()
127 nullptr : m_InputLayerNormWeightsTensor.get(), in NeonLstmFloatWorkload()
234 … InitializeArmComputeTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in NeonLstmFloatWorkload()
411 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | NeonQLstmWorkload.hpp | 50 std::unique_ptr<arm_compute::Tensor> m_InputLayerNormWeightsTensor; member in armnn::NeonQLstmWorkload
|
| D | NeonLstmFloatWorkload.hpp | 52 std::unique_ptr<arm_compute::Tensor> m_InputLayerNormWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 230 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
233 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in NeonUnidirectionalSequenceLstmFloatWorkload()
245 …rmWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeightsTensor.get(); in NeonUnidirectionalSequenceLstmFloatWorkload()
394 … InitializeArmComputeTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
900 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 242 m_InputLayerNormWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
245 …BuildArmComputeTensor(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights->GetTensorInf… in NeonUnidirectionalSequenceLstmWorkload()
257 …rmWeightTensor = m_Data.m_Parameters.m_CifgEnabled ? nullptr : m_InputLayerNormWeightsTensor.get(); in NeonUnidirectionalSequenceLstmWorkload()
375 … InitializeArmComputeTensorData(*m_InputLayerNormWeightsTensor, m_Data.m_InputLayerNormWeights); in NeonUnidirectionalSequenceLstmWorkload()
873 FreeTensorIfUnused(m_InputLayerNormWeightsTensor); in FreeUnusedTensors()
|
| D | NeonUnidirectionalSequenceLstmWorkload.hpp | 61 std::unique_ptr<arm_compute::Tensor> m_InputLayerNormWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 65 std::unique_ptr<arm_compute::Tensor> m_InputLayerNormWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefQLstmWorkload.cpp | 40 … , m_InputLayerNormWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputLayerNormWeights)) in RefQLstmWorkload()
248 … inputLayerNormWeightsDecoder = MakeDecoder<float>(m_InputLayerNormWeightsTensor->GetTensorInfo(), in Execute()
249 … m_InputLayerNormWeightsTensor->GetConstTensor<void>()); in Execute()
253 … m_InputLayerNormWeightsTensor->GetTensorInfo().GetQuantizationScale() / 1024, 0); in Execute()
338 … m_InputLayerNormWeightsTensor->GetTensorInfo().GetQuantizationScale() * in Execute()
|
| D | RefQLstmWorkload.hpp | 48 std::unique_ptr<ScopedTensorHandle> m_InputLayerNormWeightsTensor; member in armnn::RefQLstmWorkload
|