| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 215 std::initializer_list<float> output_gate_bias_; member in tflite::__anon400bb9100111::LstmOpTest
285 output_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
330 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
358 output_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
398 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
520 output_gate_bias_ = {0.046159424, -0.0012809046, 0.03563469, 0.12648113, in TEST_P()
961 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
997 output_gate_bias_ = {0.05, -0.01, 0.2, 0.1}; in TEST_F()
1047 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_F()
1073 output_gate_bias_ = {0.05, -0.01, 0.2, 0.1}; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 89 output_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
193 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias()
252 int output_gate_bias_; member in tflite::__anon67ed54f90111::UnidirectionalLSTMOpModel
371 std::vector<float> output_gate_bias_; member in tflite::__anon67ed54f90111::BaseUnidirectionalLstmTest
465 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
550 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
612 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
679 lstm.SetOutputGateBias(output_gate_bias_); in TEST_P()
742 lstm.SetOutputGateBias(output_gate_bias_); in TEST_P()
769 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
[all …]
|
| D | lstm_test.cc | 89 output_gate_bias_ = AddInput({TensorType_FLOAT32, {n_cell}}); in LSTMOpModel()
220 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias()
266 int output_gate_bias_; member in tflite::__anonb3707f480111::LSTMOpModel
321 std::vector<float> output_gate_bias_; member in tflite::__anonb3707f480111::LstmOpTest
392 lstm->SetOutputGateBias(output_gate_bias_); in SetAllWeightsAndBiases()
442 output_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
518 output_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
683 output_gate_bias_ = {0.046159424, -0.0012809046, 0.03563469, 0.12648113, in TEST_P()
1181 output_gate_bias_ = {0.05, -0.01, 0.2, 0.1}; in TEST_P()
1266 output_gate_bias_ = {0.05, -0.01, 0.2, 0.1}; in TEST_P()
[all …]
|
| D | optional_tensor_test.cc | 82 output_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
168 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias()
213 int output_gate_bias_; member in tflite::__anon67d22bc80111::LSTMOpModel
|
| D | lstm_eval_test.cc | 371 PackWeightToTensor(&output_gate_bias_tensor_, output_gate_bias_, in GetOutputBias()
373 output_gate_bias_tensor_.data.i32 = output_gate_bias_.data(); in GetOutputBias()
561 std::vector<int32_t> output_gate_bias_ = { member in tflite::__anonb312a86f0111::QuantizedLstmParam
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3145 output_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
3251 PopulateTensor(output_gate_bias_, f); in SetOutputGateBias()
3309 int output_gate_bias_; member in tflite::__anon0c68fcb70111::LSTMOpModel
3353 std::vector<float> output_gate_bias_; member in tflite::__anon0c68fcb70111::BaseLstmTest
3421 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
3504 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
3573 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
3598 output_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
3678 lstm.SetOutputGateBias(output_gate_bias_); in TEST_F()
3808 output_gate_bias_ = {0.046159424, -0.0012809046, 0.03563469, 0.12648113, in SetUp()
[all …]
|