Searched refs:output_gate_bias_ (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_test.cc | 90 output_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 205 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias() 252 int output_gate_bias_; member in tflite::__anonb3707f480111::LSTMOpModel 352 std::vector<float> output_gate_bias_; member in tflite::__anonb3707f480111::BaseLstmTest 416 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 495 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 553 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 612 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 638 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 714 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() [all …]
|
D | unidirectional_sequence_lstm_test.cc | 89 output_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel() 179 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias() 222 int output_gate_bias_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel 325 std::vector<float> output_gate_bias_; member in tflite::__anon67ed54f90111::BaseLstmTest 417 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 501 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 563 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 630 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 693 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 719 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() [all …]
|
D | optional_tensor_test.cc | 84 output_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 170 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias() 215 int output_gate_bias_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
D | nnapi_delegate_test.cc | 2312 output_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel() 2398 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias() 2440 int output_gate_bias_; member in tflite::__anon0c68fcb70111::LSTMOpModel 2470 std::vector<float> output_gate_bias_; member in tflite::__anon0c68fcb70111::BaseLstmTest 2534 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 2614 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 2639 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp() 2716 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F() 2846 output_gate_bias_ = {0.046159424, -0.0012809046, 0.03563469, 0.12648113, in SetUp() 3373 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
|