Searched refs:input_to_output_weights_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 57 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel() 145 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights() 233 int input_to_output_weights_; member in tflite::__anonb3707f480111::LSTMOpModel 306 SetWeights(input_to_output_weights_, f); in SetInputToOutputWeights() 348 std::vector<float> input_to_output_weights_; member in tflite::__anonb3707f480111::BaseLstmTest 410 input_to_output_weights_ = {-0.25065863, -0.28290087, 0.04613829, in SetUp() 490 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 548 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 607 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 633 input_to_output_weights_ = {0.10725588, -0.02335852, -0.55932593, in SetUp() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 56 input_to_output_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel() 135 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights() 208 int input_to_output_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 276 SetWeights(input_to_output_weights_, f); in SetInputToOutputWeights() 321 std::vector<float> input_to_output_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest 411 input_to_output_weights_ = {-0.25065863, -0.28290087, 0.04613829, in SetUp() 496 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 558 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 625 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 688 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() [all …]
|
D | optional_tensor_test.cc | 51 input_to_output_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 126 PopulateTensor(input_to_output_weights_, f); in SetInputToOutputWeights() 201 int input_to_output_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2279 input_to_output_weights_ = AddInput(weight_type); in LSTMOpModel() 2354 SetData(input_to_output_weights_, weight_type_, f); in SetInputToOutputWeights() 2426 int input_to_output_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2466 std::vector<float> input_to_output_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2528 input_to_output_weights_ = {-0.25065863, -0.28290087, 0.04613829, in SetUp() 2609 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 2634 input_to_output_weights_ = {0.10725588, -0.02335852, -0.55932593, in SetUp() 2712 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F() 2806 input_to_output_weights_ = { in SetUp() 3368 lstm.SetInputToOutputWeights(input_to_output_weights_); in TEST_F()
|