| /frameworks/ml/nn/common/operations/ |
| D | LSTMTest.cpp | 280 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
311 lstm.SetInputToInputWeights({-0.45018822, -0.02338299, -0.0870589, -0.34550029, 0.04266912, in TEST()
314 lstm.SetInputToCellWeights({-0.50013041, 0.1370284, 0.11810488, 0.2013163, -0.20583314, in TEST()
317 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, -0.31343272, -0.40032279, in TEST()
320 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, 0.40525138, 0.44272184, in TEST()
323 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST()
325 lstm.SetCellGateBias({0., 0., 0., 0.}); in TEST()
327 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST()
329 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST()
331 lstm.SetRecurrentToInputWeights({-0.0063535, -0.2042388, 0.31454784, -0.35746509, 0.28902304, in TEST()
[all …]
|
| D | LayerNormLSTMTest.cpp | 292 LayerNormLSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
331 lstm.SetInputToInputWeights({0.5, 0.6, 0.7, -0.8, -0.9, 0.1, 0.2, 0.3, -0.4, 0.5, in TEST()
334 lstm.SetInputToForgetWeights({-0.6, -0.1, 0.3, 0.2, 0.9, -0.5, -0.2, -0.4, 0.3, -0.8, in TEST()
337 lstm.SetInputToCellWeights({-0.4, -0.3, -0.2, -0.1, -0.5, 0.5, -0.2, -0.3, -0.2, -0.6, in TEST()
340 lstm.SetInputToOutputWeights({-0.8, -0.4, -0.2, -0.9, -0.1, -0.7, 0.3, -0.3, -0.8, -0.2, in TEST()
343 lstm.SetInputGateBias({0.03, 0.15, 0.22, 0.38}); in TEST()
345 lstm.SetForgetGateBias({0.1, -0.3, -0.2, 0.1}); in TEST()
347 lstm.SetCellGateBias({-0.05, 0.72, 0.25, 0.08}); in TEST()
349 lstm.SetOutputGateBias({0.05, -0.01, 0.2, 0.1}); in TEST()
351 lstm.SetRecurrentToInputWeights( in TEST()
[all …]
|
| D | QuantizedLSTMTest.cpp | 238 QuantizedLSTMOpModel* lstm) { in VerifyGoldens() argument
241 const int inputSize = lstm->inputSize(); in VerifyGoldens()
252 lstm->setInput(inputStep); in VerifyGoldens()
253 lstm->invoke(); in VerifyGoldens()
255 const int outputSize = lstm->outputSize(); in VerifyGoldens()
262 EXPECT_THAT(lstm->getOutput(), ElementsAreArray(expected)); in VerifyGoldens()
284 QuantizedLSTMOpModel lstm({ in TEST_F() local
325 lstm.setWeightsAndBiases( in TEST_F()
372 VerifyGoldens(lstmInput, lstmGoldenOutput, &lstm); in TEST_F()
|
| /frameworks/ml/nn/runtime/test/generated/spec_V1_0/ |
| D | lstm.example.cpp | 7 namespace generated_tests::lstm { namespace
304 namespace generated_tests::lstm { namespace
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1060 BidirectionalSequenceLSTM lstm(operation, operands); in executeOperation() local
1061 success = lstm.Prepare(operation, operands, &fwOutputShape, &bwOutputShape, in executeOperation()
1092 success = success && lstm.Eval(); in executeOperation()
|