| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm_test.cc | 440 BidirectionalLSTMOpModel lstm( in TEST_P() local
513 lstm.SetInputToInputWeights({-0.45018822, -0.02338299, -0.0870589, in TEST_P()
517 lstm.SetInputToCellWeights({-0.50013041, 0.1370284, 0.11810488, 0.2013163, in TEST_P()
521 lstm.SetInputToForgetWeights({0.09701663, 0.20334584, -0.50592935, in TEST_P()
525 lstm.SetInputToOutputWeights({-0.25065863, -0.28290087, 0.04613829, in TEST_P()
529 lstm.SetInputGateBias({0., 0., 0., 0.}); in TEST_P()
531 lstm.SetCellBias({0., 0., 0., 0.}); in TEST_P()
533 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST_P()
535 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST_P()
537 lstm.SetRecurrentToInputWeights( in TEST_P()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 138 UnidirectionalLSTMOpModel* lstm, float tolerance = 1e-5, in VerifyGoldens() argument
142 const int num_inputs = lstm->num_inputs(); in VerifyGoldens()
153 lstm->SetInput(((i * num_batches) + b) * num_inputs, batch_start, in VerifyGoldens()
162 lstm->SetInput(b * input_sequence_size * num_inputs, batch_start, in VerifyGoldens()
167 ASSERT_EQ(lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
169 const int num_outputs = lstm->num_outputs(); in VerifyGoldens()
191 EXPECT_THAT(lstm->GetOutput(), in VerifyGoldens()
255 UnidirectionalLSTMOpModel lstm( in TEST_F() local
290 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
291 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F()
[all …]
|
| D | lstm_test.cc | 341 void VerifyGoldens(LSTMOpModel* lstm, float tolerance) { in VerifyGoldens() argument
345 SetAllWeightsAndBiases(lstm); in VerifyGoldens()
346 lstm->ApplyDelegate(); in VerifyGoldens()
348 const int num_inputs = lstm->num_inputs(); in VerifyGoldens()
349 const int num_outputs = lstm->num_outputs(); in VerifyGoldens()
350 const int num_batches = lstm->num_batches(); in VerifyGoldens()
361 lstm->SetInput(b * num_inputs, batch_start, batch_end); in VerifyGoldens()
364 ASSERT_EQ(lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
375 EXPECT_THAT(lstm->GetOutput(), in VerifyGoldens()
383 void SetAllWeightsAndBiases(LSTMOpModel* lstm) { in SetAllWeightsAndBiases() argument
[all …]
|
| D | unidirectional_sequence_lstm.cc | 59 GetVariableInput(context, node, lstm::full::kCellStateTensor); in PopulateQuantizedLstmParams8x8_16()
64 GetOutputSafe(context, node, lstm::full::kOutputTensor, &output_tensor)); in PopulateQuantizedLstmParams8x8_16()
94 context, GetInputSafe(context, node, lstm::full::kInputTensor, &input)); in PopulateQuantizedLstmParams8x8_16()
97 context, node, lstm::full::kInputToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
101 GetInputSafe(context, node, lstm::full::kInputToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
105 lstm::full::kInputToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
110 GetInputSafe(context, node, lstm::full::kInputToOutputWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
114 context, node, lstm::full::kRecurrentToInputWeightsTensor); in PopulateQuantizedLstmParams8x8_16()
118 GetInputSafe(context, node, lstm::full::kRecurrentToForgetWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
123 GetInputSafe(context, node, lstm::full::kRecurrentToCellWeightsTensor, in PopulateQuantizedLstmParams8x8_16()
[all …]
|
| D | optional_tensor_test.cc | 235 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
266 lstm.SetInputToCellWeights({-0.49770179, -0.27711356, -0.09624726, 0.05100781, in TEST()
270 lstm.SetInputToForgetWeights({-0.55291498, -0.42866567, 0.13056988, in TEST()
274 lstm.SetInputToOutputWeights({0.10725588, -0.02335852, -0.55932593, in TEST()
278 lstm.SetCellBias({0., 0., 0., 0.}); in TEST()
280 lstm.SetForgetGateBias({1., 1., 1., 1.}); in TEST()
282 lstm.SetOutputGateBias({0., 0., 0., 0.}); in TEST()
284 lstm.SetRecurrentToCellWeights( in TEST()
290 lstm.SetRecurrentToForgetWeights( in TEST()
295 lstm.SetRecurrentToOutputWeights( in TEST()
[all …]
|
| D | quant_basic_lstm_test.cc | 168 QuantizedLSTMOpModel* lstm) { in VerifyGoldens() argument
171 const int inputSize = lstm->inputSize(); in VerifyGoldens()
182 lstm->setInput(inputStep); in VerifyGoldens()
183 ASSERT_EQ(lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
185 const int outputSize = lstm->outputSize(); in VerifyGoldens()
192 EXPECT_THAT(lstm->getOutput(), ElementsAreArray(expected)); in VerifyGoldens()
210 QuantizedLSTMOpModel lstm( in TEST_P() local
277 VerifyGoldens(lstmInput, lstmGoldenOutput, &lstm); in TEST_P()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | lstm_parser.cc | 108 node, tflite::ops::builtin::lstm::full::kInputToInputWeightsTensor); in HasCifg()
115 node, tflite::ops::builtin::lstm::full::kCellToForgetWeightsTensor); in HasPeephole()
121 tflite::ops::builtin::lstm::full::kForgetLayerNormCoefficientsTensor); in HasNormalization()
126 tflite::ops::builtin::lstm::full::kProjectionWeightsTensor); in HasProjection()
164 reader->AddInput(node, tflite::ops::builtin::lstm::full::kInputTensor)); in BuildLstmGate()
280 tflite::ops::builtin::lstm::full::kCellStateTensor, &cell_state)); in BuildCellStateUpdate()
406 tflite::ops::builtin::lstm::full::kProjectionWeightsTensor, in BuildOutputStateUpdate()
407 tflite::ops::builtin::lstm::full::kProjectionBiasTensor, reader, node)); in BuildOutputStateUpdate()
472 tflite::ops::builtin::lstm::full::kCellStateTensor, &old_cell_state)); in ParseLSTMAttributes()
481 tflite::ops::builtin::lstm::full::kOutputStateTensor, &old_output_state)); in ParseLSTMAttributes()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 228 void VerifyGoldens(LSTMOpModel* lstm, float tolerance) { in VerifyGoldens() argument
229 EXPECT_EQ(lstm->ApplyDelegate(), kTfLiteOk); in VerifyGoldens()
231 const int num_inputs = lstm->num_inputs(); in VerifyGoldens()
232 const int num_outputs = lstm->num_outputs(); in VerifyGoldens()
233 const int num_batches = lstm->num_batches(); in VerifyGoldens()
244 lstm->SetInput(b * num_inputs, batch_start, batch_end); in VerifyGoldens()
247 ASSERT_EQ(lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
258 EXPECT_THAT(lstm->GetOutput(), in VerifyGoldens()
318 LSTMOpModel lstm( in TEST_P() local
335 VerifyGoldens(&lstm, 0.00001f); in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 471 ops::builtin::lstm::full::kInputTensor, &input)); in lstm_eval()
474 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
479 ops::builtin::lstm::full::kInputToForgetWeightsTensor, in lstm_eval()
484 ops::builtin::lstm::full::kInputToCellWeightsTensor, in lstm_eval()
490 ops::builtin::lstm::full::kInputToOutputWeightsTensor, in lstm_eval()
494 context, node, ops::builtin::lstm::full::kRecurrentToInputWeightsTensor); in lstm_eval()
499 ops::builtin::lstm::full::kRecurrentToForgetWeightsTensor, in lstm_eval()
505 ops::builtin::lstm::full::kRecurrentToCellWeightsTensor, in lstm_eval()
511 ops::builtin::lstm::full::kRecurrentToOutputWeightsTensor, in lstm_eval()
515 context, node, ops::builtin::lstm::full::kCellToInputWeightsTensor); in lstm_eval()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 549 GetInput(context, node, ops::builtin::lstm::full::kInputTensor); in lstm_eval()
552 context, node, ops::builtin::lstm::full::kInputToInputWeightsTensor); in lstm_eval()
554 context, node, ops::builtin::lstm::full::kInputToForgetWeightsTensor); in lstm_eval()
556 context, node, ops::builtin::lstm::full::kInputToCellWeightsTensor); in lstm_eval()
558 context, node, ops::builtin::lstm::full::kInputToOutputWeightsTensor); in lstm_eval()
561 context, node, ops::builtin::lstm::full::kRecurrentToInputWeightsTensor); in lstm_eval()
563 context, node, ops::builtin::lstm::full::kRecurrentToForgetWeightsTensor); in lstm_eval()
565 context, node, ops::builtin::lstm::full::kRecurrentToCellWeightsTensor); in lstm_eval()
567 context, node, ops::builtin::lstm::full::kRecurrentToOutputWeightsTensor); in lstm_eval()
570 context, node, ops::builtin::lstm::full::kCellToInputWeightsTensor); in lstm_eval()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | lstm_test.py | 23 from tensorflow.compiler.tests import lstm
77 weights = init_weights(lstm.LSTMCellWeightsShape(num_inputs, num_nodes))
83 m, c = lstm.LSTMCell(weights, m_prev, c_prev, x, pad)
143 weights = array_ops.zeros(lstm.LSTMCellWeightsShape(num_inputs, num_nodes))
150 lstm.LSTMLayer('lstm', weights, m, c, x_seq, [pad])
152 lstm.LSTMLayer('lstm', weights, m, c, x_seq, [pad] * 2)
154 lstm.LSTMLayer('lstm', weights, m, c, x_seq, [pad] * 4)
163 weights = init_weights(lstm.LSTMCellWeightsShape(num_inputs, num_nodes))
170 out_seq = lstm.LSTMLayer('lstm', weights, m_init, c_init, x_seq, pad_seq)
243 out_seq, weights = lstm.BuildLSTMLayer(FLAGS.batch_size, FLAGS.seq_length,
|
| /external/gemmlowp/test/ |
| D | benchmark_meta_gemm.cc | 292 std::vector<Shape> lstm; in main() local
293 lstm.push_back(Shape(1, 500, 320)); in main()
294 lstm.push_back(Shape(1, 100, 500)); in main()
295 lstm.push_back(Shape(1, 500, 500)); in main()
296 lstm.push_back(Shape(1, 500, 100)); in main()
297 lstm.push_back(Shape(1, 2000, 100)); in main()
299 for (auto& shape : lstm) { in main()
327 for (auto& shape : lstm) { in main()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CudnnRNNV3.pbtxt | 58 s: "lstm"
64 s: "lstm"
183 s: "lstm"
189 s: "lstm"
315 s: "lstm"
321 s: "lstm"
|
| D | CudnnRNNParamsSize.pbtxt | 44 s: "lstm"
50 s: "lstm"
147 s: "lstm"
153 s: "lstm"
|
| D | CudnnRNNBackpropV3.pbtxt | 86 s: "lstm"
92 s: "lstm"
232 s: "lstm"
238 s: "lstm"
385 s: "lstm"
391 s: "lstm"
|
| D | CudnnRNNCanonicalToParams.pbtxt | 50 s: "lstm"
56 s: "lstm"
|
| D | CudnnRNNParamsToCanonical.pbtxt | 50 s: "lstm"
56 s: "lstm"
|
| D | CudnnRNN.pbtxt | 50 s: "lstm"
56 s: "lstm"
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3452 LSTMOpModel* lstm, float tolerance = 1e-5) { in VerifyGoldens() argument
3455 const int num_inputs = lstm->num_inputs(); in VerifyGoldens()
3464 lstm->SetInput(b * lstm->num_inputs(), batch_start, batch_end); in VerifyGoldens()
3467 ASSERT_EQ(lstm->Invoke(), kTfLiteOk); in VerifyGoldens()
3469 const int num_outputs = lstm->num_outputs(); in VerifyGoldens()
3476 EXPECT_THAT(lstm->GetOutput(), in VerifyGoldens()
3538 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F() local
3573 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
3574 lstm.SetInputToCellWeights(input_to_cell_weights_); in TEST_F()
3575 lstm.SetInputToForgetWeights(input_to_forget_weights_); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CudnnRNNParamsSize.pbtxt | 44 s: "lstm"
50 s: "lstm"
147 s: "lstm"
153 s: "lstm"
|
| D | CudnnRNNV3.pbtxt | 58 s: "lstm"
64 s: "lstm"
183 s: "lstm"
189 s: "lstm"
|
| D | CudnnRNNBackpropV3.pbtxt | 86 s: "lstm"
92 s: "lstm"
232 s: "lstm"
238 s: "lstm"
|
| D | CudnnRNNParamsToCanonical.pbtxt | 50 s: "lstm"
56 s: "lstm"
|
| D | CudnnRNNCanonicalToParams.pbtxt | 50 s: "lstm"
56 s: "lstm"
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | BUILD | 14 srcs = ["builtin_logging_ops/lstm.cc"],
15 hdrs = ["builtin_logging_ops/lstm.h"],
34 srcs = ["custom_logging_ops/lstm.cc"],
35 hdrs = ["custom_logging_ops/lstm.h"],
89 "//tensorflow/lite:testdata/lstm.bin",
|