Home
last modified time | relevance | path

Searched defs:cellStateOut (Results 1 – 25 of 27) sorted by relevance

12

/external/armnn/src/backends/neon/workloads/
DNeonQLstmWorkload.cpp146 …arm_compute::ITensor& cellStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])->GetTen… in NeonQLstmWorkload() local
246 const TensorInfo& cellStateOut, in NeonQLstmWorkloadValidate()
DNeonQuantizedLstmWorkload.cpp134 const TensorInfo& cellStateOut, in NeonQuantizedLstmWorkloadValidate()
DNeonUnidirectionalSequenceLstmWorkload.cpp48 …arm_compute::ITensor& cellStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])->GetTen… in NeonUnidirectionalSequenceLstmWorkload() local
495 const TensorInfo& cellStateOut, in NeonUnidirectionalSequenceLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp258 const TensorInfo& cellStateOut, in NeonLstmFloatWorkloadValidate()
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp514 const TensorInfo& cellStateOut, in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
/external/armnn/src/backends/cl/workloads/
DClQLstmWorkload.cpp148 …arm_compute::ICLTensor& cellStateOut = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])->GetTe… in ClQLstmWorkload() local
250 const TensorInfo& cellStateOut, in ClQLstmWorkloadValidate()
DClQuantizedLstmWorkload.cpp19 … const TensorInfo& previousOutputIn, const TensorInfo& cellStateOut, in ClQuantizedLstmWorkloadValidate()
DClLstmFloatWorkload.cpp246 … const TensorInfo& outputStateOut, const TensorInfo& cellStateOut, in ClLstmFloatWorkloadValidate()
DClUnidirectionalSequenceLstmFloatWorkload.cpp615 const TensorInfo& cellStateOut = TensorInfo(cellStateIn.GetShape(), input.GetDataType()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
/external/armnn/src/armnnSerializer/test/
DLstmSerializationTests.cpp250 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); variable
396 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); variable
561 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); variable
1582 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(0); variable
1733 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); variable
1926 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); variable
2173 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); variable
/external/armnn/src/backends/reference/workloads/
DLstm.cpp22 std::unique_ptr<Encoder<float>>& cellStateOut, in LstmImpl()
DRefLstmWorkload.cpp64 …std::unique_ptr<Encoder<float>> cellStateOut = MakeEncoder<float>(outputInfo, outputs[2]->Map()); in Execute() local
DRefUnidirectionalSequenceLstmWorkload.cpp125 …std::unique_ptr<Encoder<float>> cellStateOut = MakeEncoder<float>(cellStateInfo, cellStateOutDat… in Execute() local
/external/armnn/src/backends/neon/
DNeonLayerSupport.cpp1214 const TensorInfo& cellStateOut, in IsLstmSupported()
1361 const TensorInfo& cellStateOut, in IsQLstmSupported()
1405 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
1598 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
/external/armnn/src/armnn/
DILayerSupport.cpp866 const TensorInfo& cellStateOut, in IsLstmSupported()
1027 const TensorInfo& cellStateOut, in IsQLstmSupported()
1048 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
1223 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
DBackendHelper.cpp867 const TensorInfo& cellStateOut, in IsLstmSupported()
1121 const TensorInfo& cellStateOut, in IsQLstmSupported()
1140 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
1435 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
/external/armnn/src/backends/cl/
DClLayerSupport.cpp1192 const TensorInfo& cellStateOut, in IsLstmSupported()
1322 const TensorInfo& cellStateOut, in IsQLstmSupported()
1355 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
1547 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
/external/armnn/src/backends/backendsCommon/test/
DQLstmEndToEndTestImpl.cpp221 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(1); in QLstmEndToEnd() local
DQuantizedLstmEndToEndTestImpl.cpp123 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(0); in CreateQuantizedLstmNetwork() local
/external/android-nn-driver/
DConversionUtils_1_3.hpp519 const HalOperand* cellStateOut = GetOutputOperand<HalPolicy>(operation, 1, model); in ConvertQuantizedLstm() local
/external/armnn/src/backends/backendsCommon/
DWorkloadFactory.cpp632 …const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(2).GetTensorInfo(), dataType… in IsLayerConfigurationSupported() local
941 const TensorInfo& cellStateOut = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported() local
1035 const TensorInfo& cellStateOut = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported() local
1411 …const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(1).GetTensorInfo(), dataType… in IsLayerConfigurationSupported() local
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1749 const TensorInfo& cellStateOut, in IsLstmSupported()
2221 const TensorInfo& cellStateOut, in IsQLstmSupported()
2772 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
/external/android-nn-driver/1.0/
DHalPolicy.cpp349 const Operand* cellStateOut = GetOutputOperand<hal_1_0::HalPolicy>(operation, 2, model); in ConvertLstm() local
/external/armnn/src/armnnTestUtils/
DCreateWorkload.hpp792 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(2, "cellStateOut"); in CreateLstmWorkloadTest() local
911 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(0, "cellStateOut"); in CreateQuantizedLstmWorkloadTest() local
1075 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(1, "cellStateOut"); in CreateQLstmWorkloadTest() local
/external/armnn/src/armnn/test/
DOptimizerTests.cpp122 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(2, "cellStateOut"); in CreateLSTMLayerHelper() local

12