Searched refs:input_gate_bias_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_lstm_test.cc | 83 input_gate_bias_ = AddNullInput(); in UnidirectionalLSTMOpModel() 85 input_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel() 167 PopulateTensor(input_gate_bias_, f); in SetInputGateBias() 219 int input_gate_bias_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 322 std::vector<float> input_gate_bias_; member in tflite::__anon67ed54f90111::BaseLstmTest 414 input_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 498 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 560 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 627 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 690 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() [all …]
|
D | lstm_test.cc | 84 input_gate_bias_ = AddNullInput(); in LSTMOpModel() 86 input_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 193 PopulateTensor(input_gate_bias_, f); in SetInputGateBias() 249 int input_gate_bias_; member in tflite::__anonb3707f480111::LSTMOpModel 349 std::vector<float> input_gate_bias_; member in tflite::__anonb3707f480111::BaseLstmTest 413 input_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 492 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 550 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 609 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 944 input_gate_bias_ = {0.02234832, 0.14757581, 0.18176508, 0.10380666, in SetUp() [all …]
|
D | optional_tensor_test.cc | 78 input_gate_bias_ = AddNullInput(); in LSTMOpModel() 80 input_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 158 PopulateTensor(input_gate_bias_, f); in SetInputGateBias() 212 int input_gate_bias_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2306 input_gate_bias_ = AddNullInput(); in LSTMOpModel() 2308 input_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 2386 PopulateTensor(input_gate_bias_, f); in SetInputGateBias() 2437 int input_gate_bias_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2467 std::vector<float> input_gate_bias_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2531 input_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 2611 lstm.SetInputGateBias(input_gate_bias_); in TEST_F() 2828 input_gate_bias_ = {0.02234832, 0.14757581, 0.18176508, 0.10380666, in SetUp() 3370 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
|