Searched refs:m_BasicParameters (Results 1 – 13 of 13) sorted by relevance
27 descriptor.m_InputToForgetWeights = m_BasicParameters.m_InputToForgetWeights.get(); in CreateWorkload()28 descriptor.m_InputToCellWeights = m_BasicParameters.m_InputToCellWeights.get(); in CreateWorkload()29 descriptor.m_InputToOutputWeights = m_BasicParameters.m_InputToOutputWeights.get(); in CreateWorkload()30 descriptor.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()31 descriptor.m_RecurrentToCellWeights = m_BasicParameters.m_RecurrentToCellWeights.get(); in CreateWorkload()32 descriptor.m_RecurrentToOutputWeights = m_BasicParameters.m_RecurrentToOutputWeights.get(); in CreateWorkload()33 descriptor.m_ForgetGateBias = m_BasicParameters.m_ForgetGateBias.get(); in CreateWorkload()34 descriptor.m_CellBias = m_BasicParameters.m_CellBias.get(); in CreateWorkload()35 descriptor.m_OutputGateBias = m_BasicParameters.m_OutputGateBias.get(); in CreateWorkload()85 layer->m_BasicParameters.m_InputToForgetWeights = m_BasicParameters.m_InputToForgetWeights ? in Clone()[all …]
27 descriptor.m_InputToForgetWeights = m_BasicParameters.m_InputToForgetWeights.get(); in CreateWorkload()28 descriptor.m_InputToCellWeights = m_BasicParameters.m_InputToCellWeights.get(); in CreateWorkload()29 descriptor.m_InputToOutputWeights = m_BasicParameters.m_InputToOutputWeights.get(); in CreateWorkload()30 descriptor.m_RecurrentToForgetWeights = m_BasicParameters.m_RecurrentToForgetWeights.get(); in CreateWorkload()31 descriptor.m_RecurrentToCellWeights = m_BasicParameters.m_RecurrentToCellWeights.get(); in CreateWorkload()32 descriptor.m_RecurrentToOutputWeights = m_BasicParameters.m_RecurrentToOutputWeights.get(); in CreateWorkload()33 descriptor.m_ForgetGateBias = m_BasicParameters.m_ForgetGateBias.get(); in CreateWorkload()34 descriptor.m_CellBias = m_BasicParameters.m_CellBias.get(); in CreateWorkload()35 descriptor.m_OutputGateBias = m_BasicParameters.m_OutputGateBias.get(); in CreateWorkload()84 layer->m_BasicParameters.m_InputToForgetWeights = m_BasicParameters.m_InputToForgetWeights ? in Clone()[all …]
20 LstmBasicParameters m_BasicParameters; member in armnn::UnidirectionalSequenceLstmLayer
20 LstmBasicParameters m_BasicParameters; member in armnn::LstmLayer
83 QLstmBasicParameters m_BasicParameters; member in armnn::QLstmLayer
458 … layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle>(constTensor);459 …layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<ScopedTensorHandle>(constTensor…460 layer->m_BasicParameters.m_CellBias = std::make_unique<ScopedTensorHandle>(constTensor);461 layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<ScopedTensorHandle>(constTensor);463 layer->m_BasicParameters.m_OutputGateBias = std::make_unique<ScopedTensorHandle>(constTensor);464 …layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle>(constTe…465 …layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<ScopedTensorHandle>(constTens…466 …layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<ScopedTensorHandle>(constTensor…467 …layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<ScopedTensorHandle>(constTe…546 … layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle>(constTensor);[all …]
254 layer->m_BasicParameters.m_OutputGateBias); in CheckInputParameters()257 layer->m_BasicParameters.m_InputToForgetWeights); in CheckInputParameters()260 layer->m_BasicParameters.m_InputToCellWeights); in CheckInputParameters()263 layer->m_BasicParameters.m_InputToOutputWeights); in CheckInputParameters()266 layer->m_BasicParameters.m_RecurrentToForgetWeights); in CheckInputParameters()269 layer->m_BasicParameters.m_RecurrentToCellWeights); in CheckInputParameters()272 layer->m_BasicParameters.m_RecurrentToOutputWeights); in CheckInputParameters()275 layer->m_BasicParameters.m_ForgetGateBias); in CheckInputParameters()278 layer->m_BasicParameters.m_CellBias); in CheckInputParameters()
48 layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()50 layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()52 layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()54 layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()56 layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()58 layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()60 layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()62 layer->m_BasicParameters.m_CellBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()64 layer->m_BasicParameters.m_OutputGateBias = std::make_unique<ScopedTensorHandle> in CreateLSTMLayerHelper()67 layer->m_BasicParameters.m_InputToForgetWeights->Allocate(); in CreateLSTMLayerHelper()[all …]
318 …m_Layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()320 …m_Layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()322 …m_Layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()324 …m_Layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()326 …m_Layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()328 …m_Layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()330 …m_Layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()332 …m_Layer->m_BasicParameters.m_CellBias = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()334 …m_Layer->m_BasicParameters.m_OutputGateBias = std::make_unique<armnn::ScopedTensorHandle… in DummyLstmLayer()368 …m_Layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<armnn::ScopedTensorHandle… in DummyUnidirectionalSequenceLstmLayer()[all …]
637 … = OverrideDataType(cLayer->m_BasicParameters.m_InputToForgetWeights->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()639 … = OverrideDataType(cLayer->m_BasicParameters.m_InputToCellWeights->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()641 … = OverrideDataType(cLayer->m_BasicParameters.m_InputToOutputWeights->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()643 …= OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToForgetWeights->GetTensorInfo(), dataType… in IsLayerConfigurationSupported()645 … = OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToCellWeights->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()647 …= OverrideDataType(cLayer->m_BasicParameters.m_RecurrentToOutputWeights->GetTensorInfo(), dataType… in IsLayerConfigurationSupported()649 … = OverrideDataType(cLayer->m_BasicParameters.m_ForgetGateBias->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()651 … = OverrideDataType(cLayer->m_BasicParameters.m_CellBias->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()653 … = OverrideDataType(cLayer->m_BasicParameters.m_OutputGateBias->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()948 ARMNN_ASSERT(cLayer->m_BasicParameters.m_InputToForgetWeights.get() != nullptr); in IsLayerConfigurationSupported()[all …]
746 layer->m_BasicParameters.m_InputToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()748 layer->m_BasicParameters.m_InputToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()750 layer->m_BasicParameters.m_InputToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()752 layer->m_BasicParameters.m_RecurrentToForgetWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()754 layer->m_BasicParameters.m_RecurrentToCellWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()756 layer->m_BasicParameters.m_RecurrentToOutputWeights = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()758 layer->m_BasicParameters.m_ForgetGateBias = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()760 layer->m_BasicParameters.m_CellBias = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()762 layer->m_BasicParameters.m_OutputGateBias = std::make_unique<ScopedTensorHandle> in CreateLstmWorkloadTest()765 layer->m_BasicParameters.m_InputToForgetWeights->Allocate(); in CreateLstmWorkloadTest()[all …]
2333 layer->m_BasicParameters.m_InputToForgetWeights = in AddLstmLayer()2335 layer->m_BasicParameters.m_InputToCellWeights = in AddLstmLayer()2337 layer->m_BasicParameters.m_InputToOutputWeights = in AddLstmLayer()2339 layer->m_BasicParameters.m_RecurrentToForgetWeights = in AddLstmLayer()2341 layer->m_BasicParameters.m_RecurrentToCellWeights = in AddLstmLayer()2343 layer->m_BasicParameters.m_RecurrentToOutputWeights = in AddLstmLayer()2345 layer->m_BasicParameters.m_ForgetGateBias = in AddLstmLayer()2347 layer->m_BasicParameters.m_CellBias = in AddLstmLayer()2349 layer->m_BasicParameters.m_OutputGateBias = in AddLstmLayer()2615 layer->m_BasicParameters.m_InputToForgetWeights = in AddQLstmLayer()[all …]