Home
last modified time | relevance | path

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

/external/armnn/src/backends/neon/workloads/
DNeonQLstmWorkload.cpp140 …arm_compute::ITensor& cellStateOut = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[1])->GetTen… in NeonQLstmWorkload() local
239 const TensorInfo& cellStateOut, in NeonQLstmWorkloadValidate()
DNeonQuantizedLstmWorkload.cpp133 const TensorInfo& cellStateOut, in NeonQuantizedLstmWorkloadValidate()
DNeonLstmFloatWorkload.cpp278 const TensorInfo& cellStateOut, in NeonLstmFloatWorkloadValidate()
/external/armnn/src/backends/cl/workloads/
DClQLstmWorkload.cpp140 …arm_compute::ICLTensor& cellStateOut = static_cast<IClTensorHandle*>(m_Data.m_Outputs[1])->GetTe… in ClQLstmWorkload() local
237 const TensorInfo& cellStateOut, in ClQLstmWorkloadValidate()
DClQuantizedLstmWorkload.cpp19 … const TensorInfo& previousOutputIn, const TensorInfo& cellStateOut, in ClQuantizedLstmWorkloadValidate()
DClLstmFloatWorkload.cpp260 … const TensorInfo& outputStateOut, const TensorInfo& cellStateOut, in ClLstmFloatWorkloadValidate()
/external/armnn/src/backends/backendsCommon/test/
DQuantizedLstmEndToEndTestImpl.cpp123 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(0); in CreateQuantizedLstmNetwork() local
197 …boost::multi_array<int16_t, 2> cellStateOut = MakeTensor<int16_t, 2>(cellStateOutVectorDesc, cellS… in QuantizedLstmEndToEnd() local
DQLstmEndToEndTestImpl.cpp214 armnn::IConnectableLayer* const cellStateOut = net->AddOutputLayer(1); in QLstmEndToEnd() local
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp53 …std::unique_ptr<Encoder<float>> cellStateOut = MakeEncoder<float>(outputInfo, m_Data.m_Outputs[2… in Execute() local
/external/armnn/src/backends/neon/
DNeonLayerSupport.cpp581 const TensorInfo& cellStateOut, in IsLstmSupported()
728 const TensorInfo& cellStateOut, in IsQLstmSupported()
772 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
/external/armnn/src/backends/cl/
DClLayerSupport.cpp613 const TensorInfo& cellStateOut, in IsLstmSupported()
743 const TensorInfo& cellStateOut, in IsQLstmSupported()
776 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
/external/android-nn-driver/
DConversionUtils_1_3.hpp435 const HalOperand* cellStateOut = GetOutputOperand<HalPolicy>(operation, 1, model); in ConvertQuantizedLstm() local
DConversionUtils_1_2.hpp1828 const HalOperand* cellStateOut = GetOutputOperand<HalPolicy>(operation, 0, model); in ConvertQuantized16BitLstm() local
2456 const HalOperand* cellStateOut = GetOutputOperand<HalPolicy>(operation, 2, model); in ConvertLstm() local
/external/armnn/src/armnn/
DLayerSupport.cpp367 const TensorInfo& outputStateOut, const TensorInfo& cellStateOut, in IsLstmSupported()
504 const TensorInfo& cellStateOut, in IsQLstmSupported()
520 const TensorInfo& cellStateOut, in IsQuantizedLstmSupported()
/external/armnn/src/backends/backendsCommon/
DWorkloadFactory.cpp504 …const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(2).GetTensorInfo(), dataType… in IsLayerConfigurationSupported() local
798 const TensorInfo& cellStateOut = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported() local
889 const TensorInfo& cellStateOut = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported() local
/external/android-nn-driver/1.0/
DHalPolicy.cpp346 const Operand* cellStateOut = GetOutputOperand<hal_1_0::HalPolicy>(operation, 2, model); in ConvertLstm() local
/external/armnn/src/armnn/test/
DCreateWorkload.hpp741 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(2, "cellStateOut"); in CreateLstmWorkloadTest() local
860 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(0, "cellStateOut"); in CreateQuantizedLstmWorkloadTest() local
1024 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(1, "cellStateOut"); in CreateQLstmWorkloadTest() local
DOptimizerTests.cpp121 Layer* const cellStateOut = graph.AddLayer<OutputLayer>(2, "cellStateOut"); in CreateLSTMLayerHelper() local
/external/armnn/src/armnnSerializer/test/
DSerializerTests.cpp3148 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); in BOOST_AUTO_TEST_CASE() local
3294 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); in BOOST_AUTO_TEST_CASE() local
3459 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(2); in BOOST_AUTO_TEST_CASE() local
4422 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(0); in BOOST_AUTO_TEST_CASE() local
4643 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); in BOOST_AUTO_TEST_CASE() local
4831 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); in BOOST_AUTO_TEST_CASE() local
5072 armnn::IConnectableLayer* const cellStateOut = network->AddOutputLayer(1); in BOOST_AUTO_TEST_CASE() local
/external/armnn/src/backends/reference/
DRefLayerSupport.cpp1182 const TensorInfo& cellStateOut, in IsLstmSupported()
1638 const TensorInfo& cellStateOut, in IsQLstmSupported()