Home
last modified time | relevance | path

Searched defs:cellStateIn (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/armnn/src/armnnSerializer/test/
H A DLstmSerializationTests.cpp244 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
390 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
555 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
1578 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
1728 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(2); variable
1921 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(2); variable
2168 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(2); variable
2291 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
2432 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
2592 armnn::IConnectableLayer* const cellStateIn = network->AddInputLayer(1); variable
[all …]
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQLstmWorkload.cpp143 …const arm_compute::ITensor& cellStateIn = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[2])->Ge… in NeonQLstmWorkload() local
244 const TensorInfo& cellStateIn, in NeonQLstmWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp45 …const arm_compute::ITensor& cellStateIn = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[2])->Ge… in NeonUnidirectionalSequenceLstmWorkload() local
493 const TensorInfo& cellStateIn, in NeonUnidirectionalSequenceLstmWorkloadValidate()
H A DNeonQuantizedLstmWorkload.cpp132 const TensorInfo& cellStateIn, in NeonQuantizedLstmWorkloadValidate()
H A DNeonLstmFloatWorkload.cpp255 const TensorInfo& cellStateIn, in NeonLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp512 const TensorInfo& cellStateIn, in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQLstmWorkload.cpp145 …arm_compute::ICLTensor& cellStateIn = static_cast<IClTensorHandle*>(m_Data.m_Inputs[2])->G… in ClQLstmWorkload() local
248 const TensorInfo& cellStateIn, in ClQLstmWorkloadValidate()
H A DClLstmFloatWorkload.cpp245 … const TensorInfo& cellStateIn, const TensorInfo& scratchBuffer, in ClLstmFloatWorkloadValidate()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp510 const TensorInfo& cellStateIn, in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonLayerSupport.cpp1211 const TensorInfo& cellStateIn, in IsLstmSupported()
1403 const TensorInfo& cellStateIn, in IsQuantizedLstmSupported()
1596 const TensorInfo& cellStateIn, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DLstm.cpp20 std::unique_ptr<Decoder<float>>& cellStateIn, in LstmImpl()
H A DRefLstmWorkload.cpp72 std::unique_ptr<Decoder<float>> cellStateIn = MakeDecoder<float>(inputInfo, inputs[2]->Map()); in Execute() local
H A DRefUnidirectionalSequenceLstmWorkload.cpp142 … std::unique_ptr<Decoder<float>> cellStateIn = MakeDecoder<float>(cellStateInfo, cellStateInData); in Execute() local
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp789 Layer* const cellStateIn = graph.AddLayer<InputLayer>(2, "cellStateIn"); in CreateLstmWorkloadTest() local
908 Layer* const cellStateIn = graph.AddLayer<InputLayer>(1, "cellStateIn"); in CreateQuantizedLstmWorkloadTest() local
1072 Layer* const cellStateIn = graph.AddLayer<InputLayer>(2, "cellStateIn"); in CreateQLstmWorkloadTest() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DQLstmEndToEndTestImpl.cpp216 armnn::IConnectableLayer* const cellStateIn = net->AddInputLayer(2); in QLstmEndToEnd() local
H A DQuantizedLstmEndToEndTestImpl.cpp120 armnn::IConnectableLayer* const cellStateIn = net->AddInputLayer(1); in CreateQuantizedLstmNetwork() local
/aosp_15_r20/external/armnn/src/armnn/
H A DILayerSupport.cpp863 const TensorInfo& cellStateIn, in IsLstmSupported()
1221 const TensorInfo& cellStateIn, in IsUnidirectionalSequenceLstmSupported()
H A DBackendHelper.cpp864 const TensorInfo& cellStateIn, in IsLstmSupported()
1433 const TensorInfo& cellStateIn, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClLayerSupport.cpp1189 const TensorInfo& cellStateIn, in IsLstmSupported()
1545 const TensorInfo& cellStateIn, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp166 …LayerInputHandle cellStateIn = ConvertToLayerInputHandle<hal_1_0::HalPolicy>(operation, 19, model,… in ConvertLstm() local
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefLayerSupport.cpp1746 const TensorInfo& cellStateIn, in IsLstmSupported()
2770 const TensorInfo& cellStateIn, in IsUnidirectionalSequenceLstmSupported()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactory.cpp627 …const TensorInfo& cellStateIn = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorI… in IsLayerConfigurationSupported() local
1407 …const TensorInfo& cellStateIn = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorI… in IsLayerConfigurationSupported() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_2.hpp2492 LayerInputHandle cellStateIn = ConvertToLayerInputHandle<HalPolicy>(operation, 19, model, data); in ConvertLstm() local
3139 LayerInputHandle cellStateIn = ConvertToLayerInputHandle<HalPolicy>(operation, 19, model, data); in ConvertUnidirectionalSequenceLstm() local
/aosp_15_r20/external/armnn/src/armnn/test/
H A DOptimizerTests.cpp119 Layer* const cellStateIn = graph.AddLayer<InputLayer>(2, "cellStateIn"); in CreateLSTMLayerHelper() local
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp2677 ANeuralNetworksOperandType cellStateIn = floatTensor2D; in TEST() local
2761 ANeuralNetworksOperandType cellStateIn = floatTensor2D; in lstmTestV1_2() local
4173 ANeuralNetworksOperandType cellStateIn = {.type = inputOperandCode, in unidirectionalSequenceLSTMTest() local

12