| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 214 std::initializer_list<float> cell_gate_bias_; member in tflite::__anonb664ae370111::LstmOpTest
283 cell_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
330 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
356 cell_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
398 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
514 cell_gate_bias_ = {-0.024379363, 0.0055531194, 0.23377132, 0.033463873, in TEST_P()
961 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_P()
995 cell_gate_bias_ = {-0.05, 0.72, 0.25, 0.08}; in TEST_F()
1047 forget_gate_bias_, cell_gate_bias_, output_gate_bias_, in TEST_F()
1072 cell_gate_bias_ = {-0.05, 0.72, 0.25, 0.08}; in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 88 cell_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
189 PopulateTensor(cell_gate_bias_, f); in SetCellBias()
251 int cell_gate_bias_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
369 std::vector<float> cell_gate_bias_; member in tflite::__anond8582fe00111::BaseUnidirectionalLstmTest
463 cell_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
548 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
610 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
677 lstm.SetCellBias(cell_gate_bias_); in TEST_P()
740 lstm.SetCellBias(cell_gate_bias_); in TEST_P()
767 cell_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
[all …]
|
| D | lstm_test.cc | 88 cell_gate_bias_ = AddInput({TensorType_FLOAT32, {n_cell}}); in LSTMOpModel()
216 PopulateTensor(cell_gate_bias_, f); in SetCellBias()
265 int cell_gate_bias_; member in tflite::__anonfa79752f0111::LSTMOpModel
319 std::vector<float> cell_gate_bias_; member in tflite::__anonfa79752f0111::LstmOpTest
390 lstm->SetCellBias(cell_gate_bias_); in SetAllWeightsAndBiases()
440 cell_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
516 cell_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
677 cell_gate_bias_ = {-0.024379363, 0.0055531194, 0.23377132, 0.033463873, in TEST_P()
1179 cell_gate_bias_ = {-0.05, 0.72, 0.25, 0.08}; in TEST_P()
1265 cell_gate_bias_ = {-0.05, 0.72, 0.25, 0.08}; in TEST_P()
[all …]
|
| D | optional_tensor_test.cc | 81 cell_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
164 PopulateTensor(cell_gate_bias_, f); in SetCellBias()
212 int cell_gate_bias_; member in tflite::__anon4dd1434f0111::LSTMOpModel
|
| D | lstm_eval_test.cc | 365 PackWeightToTensor(&cell_gate_bias_tensor_, cell_gate_bias_, in GetCellBias()
367 cell_gate_bias_tensor_.data.i32 = cell_gate_bias_.data(); in GetCellBias()
556 std::vector<int32_t> cell_gate_bias_ = { member in tflite::__anon39d466b60111::QuantizedLstmParam
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3351 std::vector<float> cell_gate_bias_; member in tflite::__anonde5f765e0111::BaseLstmTest
3419 cell_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
3502 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
3571 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
3596 cell_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
3676 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
3802 cell_gate_bias_ = {-0.024379363, 0.0055531194, 0.23377132, 0.033463873, in SetUp()
4336 lstm.SetCellBias(cell_gate_bias_); in TEST_F()
4377 cell_gate_bias_ = {-0.05, 0.72, 0.25, 0.08}; in SetUp()
4476 layer_norm_lstm.SetCellBias(cell_gate_bias_); in TEST_F()
[all …]
|