| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 213 std::initializer_list<float> forget_gate_bias_; member in tflite::__anonb664ae370111::LstmOpTest
284 forget_gate_bias_ = {1., 1., 1., 1.}; in TEST_P()
330 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
357 forget_gate_bias_ = {1., 1., 1., 1.}; in TEST_P()
398 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
508 forget_gate_bias_ = {0.035185695, -0.042891346, -0.03032477, 0.23027696, in TEST_P()
961 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
993 forget_gate_bias_ = {0.1, -0.3, -0.2, 0.1}; in TEST_F()
1047 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_F()
1071 forget_gate_bias_ = {0.1, -0.3, -0.2, 0.1}; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 87 forget_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
185 PopulateTensor(forget_gate_bias_, f); in SetForgetGateBias()
250 int forget_gate_bias_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
370 std::vector<float> forget_gate_bias_; member in tflite::__anond8582fe00111::BaseUnidirectionalLstmTest
464 forget_gate_bias_ = {1., 1., 1., 1.}; in SetUp()
549 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
611 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
678 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_P()
741 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_P()
768 forget_gate_bias_ = {1., 1., 1., 1.}; in SetUp()
[all …]
|
| D | lstm_test.cc | 87 forget_gate_bias_ = AddInput({TensorType_FLOAT32, {n_cell}}); in LSTMOpModel()
212 PopulateTensor(forget_gate_bias_, f); in SetForgetGateBias()
264 int forget_gate_bias_; member in tflite::__anonfa79752f0111::LSTMOpModel
320 std::vector<float> forget_gate_bias_; member in tflite::__anonfa79752f0111::LstmOpTest
391 lstm->SetForgetGateBias(forget_gate_bias_); in SetAllWeightsAndBiases()
441 forget_gate_bias_ = {1., 1., 1., 1.}; in TEST_P()
517 forget_gate_bias_ = {1., 1., 1., 1.}; in TEST_P()
671 forget_gate_bias_ = {0.035185695, -0.042891346, -0.03032477, 0.23027696, in TEST_P()
1177 forget_gate_bias_ = {0.1, -0.3, -0.2, 0.1}; in TEST_P()
1264 forget_gate_bias_ = {0.1, -0.3, -0.2, 0.1}; in TEST_P()
[all …]
|
| D | optional_tensor_test.cc | 80 forget_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
160 PopulateTensor(forget_gate_bias_, f); in SetForgetGateBias()
211 int forget_gate_bias_; member in tflite::__anon4dd1434f0111::LSTMOpModel
|
| D | lstm_eval_test.cc | 359 PackWeightToTensor(&forget_gate_bias_tensor_, forget_gate_bias_, in GetForgetBias()
361 forget_gate_bias_tensor_.data.i32 = forget_gate_bias_.data(); in GetForgetBias()
551 std::vector<int32_t> forget_gate_bias_ = { member in tflite::__anon39d466b60111::QuantizedLstmParam
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3143 forget_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
3243 PopulateTensor(forget_gate_bias_, f); in SetForgetGateBias()
3307 int forget_gate_bias_; member in tflite::__anonde5f765e0111::LSTMOpModel
3352 std::vector<float> forget_gate_bias_; member in tflite::__anonde5f765e0111::BaseLstmTest
3420 forget_gate_bias_ = {1., 1., 1., 1.}; in SetUp()
3503 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
3572 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
3597 forget_gate_bias_ = {1., 1., 1., 1.}; in SetUp()
3677 lstm.SetForgetGateBias(forget_gate_bias_); in TEST_F()
3796 forget_gate_bias_ = {0.035185695, -0.042891346, -0.03032477, 0.23027696, in SetUp()
[all …]
|