Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A 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
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQLstmWorkload.cpp146 …arm_compute::ITensor& cellStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])->GetTen… in NeonQLstmWorkload() local
246 const TensorInfo& cellStateOut, in NeonQLstmWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp48 …arm_compute::ITensor& cellStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])->GetTen… in NeonUnidirectionalSequenceLstmWorkload() local
495 const TensorInfo& cellStateOut, in NeonUnidirectionalSequenceLstmWorkloadValidate()
H A DNeonQuantizedLstmWorkload.cpp134 const TensorInfo& cellStateOut, in NeonQuantizedLstmWorkloadValidate()
H A DNeonLstmFloatWorkload.cpp258 const TensorInfo& cellStateOut, in NeonLstmFloatWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQLstmWorkload.cpp148 …arm_compute::ICLTensor& cellStateOut = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])->GetTe… in ClQLstmWorkload() local
250 const TensorInfo& cellStateOut, in ClQLstmWorkloadValidate()
H A DClQuantizedLstmWorkload.cpp19 … const TensorInfo& previousOutputIn, const TensorInfo& cellStateOut, in ClQuantizedLstmWorkloadValidate()
H A DClLstmFloatWorkload.cpp246 … const TensorInfo& outputStateOut, const TensorInfo& cellStateOut, in ClLstmFloatWorkloadValidate()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp615 const TensorInfo& cellStateOut = TensorInfo(cellStateIn.GetShape(), input.GetDataType()); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp305 float* cellStateOut; in execute() local
357 _Float16* cellStateOut; in execute() local
/aosp_15_r20/external/armnn/src/armnn/
H A 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()
H A 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()
/aosp_15_r20/external/armnn/src/backends/neon/
H A 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()
/aosp_15_r20/external/armnn/src/backends/cl/
H A 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()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A 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
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefLayerSupport.cpp1749 const TensorInfo& cellStateOut, in IsLstmSupported()
2221 const TensorInfo& cellStateOut, in IsQLstmSupported()
2772 const TensorInfo& cellStateOut, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp22 std::unique_ptr<Encoder<float>>& cellStateOut, in LstmImpl()
H A DRefLstmWorkload.cpp64 …std::unique_ptr<Encoder<float>> cellStateOut = MakeEncoder<float>(outputInfo, outputs[2]->Map()); in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp125 …std::unique_ptr<Encoder<float>> cellStateOut = MakeEncoder<float>(cellStateInfo, cellStateOutDat… in Execute() local
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A 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
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp221 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(1); in QLstmEndToEnd() local
H A DQuantizedLstmEndToEndTestImpl.cpp123 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(0); in CreateQuantizedLstmNetwork() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_3.hpp519 const HalOperand* cellStateOut = GetOutputOperand<HalPolicy>(operation, 1, model); in ConvertQuantizedLstm() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation() local
1620 RunTimeOperandInfo& cellStateOut = in executeOperation() local
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp349 const Operand* cellStateOut = GetOutputOperand<hal_1_0::HalPolicy>(operation, 2, model); in ConvertLstm() local

12