Searched refs:lstm_layer (Results 1 – 3 of 3) sorted by relevance
327 std::unique_ptr<arm_compute::NELSTMLayer> lstm_layer(new arm_compute::NELSTMLayer()); in NeonUnidirectionalSequenceLstmFloatWorkload() local328 lstm_layer->configure(inputLSTM, in NeonUnidirectionalSequenceLstmFloatWorkload()349 m_Layers.emplace_back(std::move(lstm_layer)); in NeonUnidirectionalSequenceLstmFloatWorkload()
313 std::unique_ptr<arm_compute::NEQLSTMLayer> lstm_layer(new arm_compute::NEQLSTMLayer()); in NeonUnidirectionalSequenceLstmWorkload() local315 lstm_layer->configure(inputLSTM, in NeonUnidirectionalSequenceLstmWorkload()332 m_Layers.emplace_back(std::move(lstm_layer)); in NeonUnidirectionalSequenceLstmWorkload()
323 std::unique_ptr<arm_compute::CLLSTMLayer> lstm_layer(new arm_compute::CLLSTMLayer()); in ClUnidirectionalSequenceLstmFloatWorkload() local324 lstm_layer->configure(clCompileContext, in ClUnidirectionalSequenceLstmFloatWorkload()346 m_Layers.emplace_back(std::move(lstm_layer)); in ClUnidirectionalSequenceLstmFloatWorkload()