Home
last modified time | relevance | path

Searched defs:outputStateOut (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQLstmWorkload.cpp145 …arm_compute::ITensor& outputStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTen… in NeonQLstmWorkload() local
247 const TensorInfo& outputStateOut, in NeonQLstmWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp47 …arm_compute::ITensor& outputStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[0])->GetTen… in NeonUnidirectionalSequenceLstmWorkload() local
494 const TensorInfo& outputStateOut, in NeonUnidirectionalSequenceLstmWorkloadValidate()
H A DNeonQuantizedLstmWorkload.cpp135 const TensorInfo& outputStateOut, in NeonQuantizedLstmWorkloadValidate()
H A DNeonLstmFloatWorkload.cpp257 const TensorInfo& outputStateOut, in NeonLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp513 const TensorInfo& outputStateOut, in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQLstmWorkload.cpp147 …arm_compute::ICLTensor& outputStateOut = static_cast<IClTensorHandle*>(m_Data.m_Outputs[0])->GetTe… in ClQLstmWorkload() local
251 const TensorInfo& outputStateOut, in ClQLstmWorkloadValidate()
H A DClLstmFloatWorkload.cpp246const TensorInfo& outputStateOut, const TensorInfo& cellStateOut, in ClLstmFloatWorkloadValidate()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp614 const TensorInfo& outputStateOut = TensorInfo(outputStateIn.GetShape(), input.GetDataType()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp249 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(1); variable
395 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(1); variable
560 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(1); variable
1732 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(0); variable
1925 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(0); variable
2172 armnn::IConnectableLayer* const outputStateOut = network->AddOutputLayer(0); variable
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp304 float* outputStateOut; in execute() local
356 _Float16* outputStateOut; in execute() local
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonLayerSupport.cpp1213 const TensorInfo& outputStateOut, in IsLstmSupported()
1360 const TensorInfo& outputStateOut, in IsQLstmSupported()
1406 const TensorInfo& outputStateOut, in IsQuantizedLstmSupported()
1597 const TensorInfo& outputStateOut, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/armnn/
H A DILayerSupport.cpp865 const TensorInfo& outputStateOut, in IsLstmSupported()
1026 const TensorInfo& outputStateOut, in IsQLstmSupported()
1222 const TensorInfo& outputStateOut, in IsUnidirectionalSequenceLstmSupported()
H A DBackendHelper.cpp866 const TensorInfo& outputStateOut, in IsLstmSupported()
1120 const TensorInfo& outputStateOut, in IsQLstmSupported()
1434 const TensorInfo& outputStateOut, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClLayerSupport.cpp1191 const TensorInfo& outputStateOut, in IsLstmSupported()
1321 const TensorInfo& outputStateOut, in IsQLstmSupported()
1546 const TensorInfo& outputStateOut, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefLayerSupport.cpp1748 const TensorInfo& outputStateOut, in IsLstmSupported()
2220 const TensorInfo& outputStateOut, in IsQLstmSupported()
2771 const TensorInfo& outputStateOut, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp21 std::unique_ptr<Encoder<float>>& outputStateOut, in LstmImpl()
H A DRefLstmWorkload.cpp63 …std::unique_ptr<Encoder<float>> outputStateOut = MakeEncoder<float>(outputInfo, outputs[1]->Map()); in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp124 …std::unique_ptr<Encoder<float>> outputStateOut = MakeEncoder<float>(outputStateInfo, outputStateOu… in Execute() local
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp791 Layer* const outputStateOut = graph.AddLayer<OutputLayer>(1, "outputStateOut"); in CreateLstmWorkloadTest() local
912 Layer* const outputStateOut = graph.AddLayer<OutputLayer>(1, "outputStateOut"); in CreateQuantizedLstmWorkloadTest() local
1074 Layer* const outputStateOut = graph.AddLayer<OutputLayer>(0, "outputStateOut"); in CreateQLstmWorkloadTest() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp220 armnn::IConnectableLayer* const outputStateOut = net->AddOutputLayer(0); in QLstmEndToEnd() local
H A DQuantizedLstmEndToEndTestImpl.cpp124 armnn::IConnectableLayer* const outputStateOut = net->AddOutputLayer(1); in CreateQuantizedLstmNetwork() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactory.cpp631 …const TensorInfo& outputStateOut = OverrideDataType(layer.GetOutputSlot(1).GetTensorInfo(), dataTy… in IsLayerConfigurationSupported() local
940 const TensorInfo& outputStateOut = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported() local
1410 …const TensorInfo& outputStateOut = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataTy… in IsLayerConfigurationSupported() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_3.hpp512 const HalOperand* outputStateOut = GetOutputOperand<HalPolicy>(operation, 0, model); in ConvertQuantizedLstm() local
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp343 const Operand* outputStateOut = GetOutputOperand<hal_1_0::HalPolicy>(operation, 1, model); in ConvertLstm() local
/aosp_15_r20/external/armnn/src/armnn/test/
H A DOptimizerTests.cpp121 Layer* const outputStateOut = graph.AddLayer<OutputLayer>(1, "outputStateOut"); in CreateLSTMLayerHelper() local

12