/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
|