| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | lstm_full_test.cc | 201 std::initializer_list<float> input_to_input_weights_; member in tflite::__anone147aac60111::LstmOpTest
273 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, -0.34550029, in TEST_P()
325 input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in TEST_P()
393 input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in TEST_P()
414 input_to_input_weights_ = { in TEST_P()
956 input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in TEST_P()
975 input_to_input_weights_ = {0.5, 0.6, 0.7, -0.8, -0.9, 0.1, 0.2, in TEST_F()
1042 input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in TEST_F()
1105 input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_test.cc | 51 input_to_input_weights_ = AddNullInput(); in LSTMOpModel()
53 input_to_input_weights_ = AddInput({weight_type, {n_cell, n_input}}); in LSTMOpModel()
148 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights()
244 int input_to_input_weights_; member in tflite::__anon299f347e0111::LSTMOpModel
314 std::vector<float> input_to_input_weights_; member in tflite::__anon299f347e0111::LstmOpTest
384 lstm->SetInputToInputWeights(input_to_input_weights_); in SetAllWeightsAndBiases()
430 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, -0.34550029, in TEST_P()
577 input_to_input_weights_ = { in TEST_P()
1159 input_to_input_weights_ = {0.5, 0.6, 0.7, -0.8, -0.9, 0.1, 0.2, in TEST_P()
1327 input_to_input_weights_ = AddNullInput(); in LSTMIntegerOpModel()
[all …]
|
| H A D | unidirectional_sequence_lstm_test.cc | 58 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights()
112 std::vector<float> input_to_input_weights_; member in tflite::__anon47ec8c2f0111::BaseUnidirectionalLstmTest
199 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, in SetUp()
290 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
352 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
419 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_P()
482 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_P()
739 input_to_input_weights_ = { in SetUp()
1379 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
1449 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_P()
[all …]
|
| H A D | optional_tensor_test.cc | 42 input_to_input_weights_ = AddNullInput(); in LSTMOpModel()
44 input_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
112 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights()
196 int input_to_input_weights_; member in tflite::__anon499b733e0111::LSTMOpModel
|
| H A D | unidirectional_sequence_lstm_test_util.h | 47 input_to_input_weights_ = AddNullInput(); in n_batch_()
49 input_to_input_weights_ = AddInput(weights_type); in n_batch_()
135 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights()
233 int input_to_input_weights_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 3182 input_to_input_weights_ = AddNullInput(); in LSTMOpModel()
3184 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel()
3272 SetData(input_to_input_weights_, weight_type_, f); in SetInputToInputWeights()
3369 int input_to_input_weights_; member in tflite::__anon1a74836d0111::LSTMOpModel
3423 std::vector<float> input_to_input_weights_; member in tflite::__anon1a74836d0111::BaseLstmTest
3484 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, in SetUp()
3573 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
3642 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
3769 input_to_input_weights_ = { in SetUp()
4407 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
[all …]
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 57 input_to_input_weights_ = in LSTMCell()
361 input_, input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in Prepare()
996 GetBuffer<const float>(input_to_input_weights_), in Eval()
1031 GetOptionalBuffer<const _Float16>(input_to_input_weights_), in Eval()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 208 const RunTimeOperandInfo* input_to_input_weights_; variable
|