| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTMTest.cpp | 279 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
310 lstm.SetInputToInputWeights({-0.45018822, -0.02338299, -0.0870589, -0.34550029, 0.04266912, in TEST()
313 lstm.SetInputToCellWeights({-0.50013041, 0.1370284, 0.11810488, 0.2013163, -0.20583314, in TEST()
316 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, in TEST()
319 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, 0.40525138, 0.44272184, in TEST()
322 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST()
324 lstm.SetCellGateBias({0., 0., 0., 0.}); in TEST()
326 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST()
328 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST()
330 lstm.SetRecurrentToInputWeights({-0.0063535, -0.2042388, 0.31454784, -0.35746509, 0.28902304, in TEST()
[all …]
|
| D | LayerNormLSTMTest.cpp | 293 LayerNormLSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
332 lstm.SetInputToInputWeights({0.5, 0.6, 0.7, -0.8, -0.9, 0.1, 0.2, 0.3, -0.4, 0.5, in TEST()
335 lstm.SetInputToForgetWeights({-0.6, -0.1, 0.3, 0.2, 0.9, -0.5, -0.2, -0.4, 0.3, -0.8, in TEST()
338 lstm.SetInputToCellWeights({-0.4, -0.3, -0.2, -0.1, -0.5, 0.5, -0.2, -0.3, -0.2, -0.6, in TEST()
341 lstm.SetInputToOutputWeights({-0.8, -0.4, -0.2, -0.9, -0.1, -0.7, 0.3, -0.3, -0.8, -0.2, in TEST()
344 lstm.SetInputGateBias({0.03, 0.15, 0.22, 0.38}); in TEST()
346 lstm.SetForgetGateBias({0.1, -0.3, -0.2, 0.1}); in TEST()
348 lstm.SetCellGateBias({-0.05, 0.72, 0.25, 0.08}); in TEST()
350 lstm.SetOutputGateBias({0.05, -0.01, 0.2, 0.1}); in TEST()
352 lstm.SetRecurrentToInputWeights( in TEST()
[all …]
|
| D | QuantizedLSTMTest.cpp | 239 QuantizedLSTMOpModel* lstm) { in VerifyGoldens() argument
242 const int inputSize = lstm->inputSize(); in VerifyGoldens()
253 lstm->setInput(inputStep); in VerifyGoldens()
254 lstm->invoke(); in VerifyGoldens()
256 const int outputSize = lstm->outputSize(); in VerifyGoldens()
263 EXPECT_THAT(lstm->getOutput(), ElementsAreArray(expected)); in VerifyGoldens()
285 QuantizedLSTMOpModel lstm({ in TEST_F() local
326 lstm.setWeightsAndBiases( in TEST_F()
373 VerifyGoldens(lstmInput, lstmGoldenOutput, &lstm); in TEST_F()
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | LSTM.cpp | 25 namespace lstm { namespace
69 NN_DEFINE_VALIDATION_FUNCTION(LSTM, lstm::validate);
|
| /packages/modules/NeuralNetworks/runtime/test/generated/spec_V1_0/ |
| D | lstm.example.cpp | 7 namespace generated_tests::lstm { namespace
304 namespace generated_tests::lstm { namespace
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1007 BidirectionalSequenceLSTM lstm(operation, operands); in executeOperation() local
1008 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1039 success = success && lstm.Eval(); in executeOperation()
|