Searched refs:input_to_input_weights_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_lstm_test.cc | 49 input_to_input_weights_ = AddNullInput(); in UnidirectionalLSTMOpModel() 51 input_to_input_weights_ = AddInput(weights_type); in UnidirectionalLSTMOpModel() 123 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights() 205 int input_to_input_weights_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 264 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights() 318 std::vector<float> input_to_input_weights_; member in tflite::__anon67ed54f90111::BaseLstmTest 403 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, in SetUp() 493 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() 555 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() 622 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() [all …]
|
D | lstm_test.cc | 50 input_to_input_weights_ = AddNullInput(); in LSTMOpModel() 52 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel() 133 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights() 230 int input_to_input_weights_; member in tflite::__anonb3707f480111::LSTMOpModel 294 SetWeights(input_to_input_weights_, f); in SetInputToInputWeights() 345 std::vector<float> input_to_input_weights_; member in tflite::__anonb3707f480111::BaseLstmTest 402 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, in SetUp() 487 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() 545 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() 604 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() [all …]
|
D | optional_tensor_test.cc | 44 input_to_input_weights_ = AddNullInput(); in LSTMOpModel() 46 input_to_input_weights_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 114 PopulateTensor(input_to_input_weights_, f); in SetInputToInputWeights() 198 int input_to_input_weights_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2272 input_to_input_weights_ = AddNullInput(); in LSTMOpModel() 2274 input_to_input_weights_ = AddInput(weight_type); in LSTMOpModel() 2342 SetData(input_to_input_weights_, weight_type_, f); in SetInputToInputWeights() 2423 int input_to_input_weights_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2463 std::vector<float> input_to_input_weights_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2520 input_to_input_weights_ = {-0.45018822, -0.02338299, -0.0870589, in SetUp() 2606 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F() 2730 input_to_input_weights_ = { in SetUp() 3365 lstm.SetInputToInputWeights(input_to_input_weights_); in TEST_F()
|