| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 212 std::initializer_list<float> input_gate_bias_; member in tflite::__anonb664ae370111::LstmOpTest
282 input_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
329 cell_to_forget_weights_, cell_to_output_weights_, input_gate_bias_, in TEST_P()
397 cell_to_forget_weights_, cell_to_output_weights_, input_gate_bias_, in TEST_P()
502 input_gate_bias_ = {0.02234832, 0.14757581, 0.18176508, 0.10380666, in TEST_P()
960 cell_to_forget_weights_, cell_to_output_weights_, input_gate_bias_, in TEST_P()
991 input_gate_bias_ = {0.03, 0.15, 0.22, 0.38}; in TEST_F()
1046 cell_to_forget_weights_, cell_to_output_weights_, input_gate_bias_, in TEST_F()
1109 cell_to_forget_weights_, cell_to_output_weights_, input_gate_bias_, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 83 input_gate_bias_ = AddNullInput(); in LSTMOpModel()
85 input_gate_bias_ = AddInput({TensorType_FLOAT32, {n_cell}}); in LSTMOpModel()
208 PopulateTensor(input_gate_bias_, f); in SetInputGateBias()
263 int input_gate_bias_; member in tflite::__anonfa79752f0111::LSTMOpModel
318 std::vector<float> input_gate_bias_; member in tflite::__anonfa79752f0111::LstmOpTest
389 lstm->SetInputGateBias(input_gate_bias_); in SetAllWeightsAndBiases()
439 input_gate_bias_ = {0., 0., 0., 0.}; in TEST_P()
665 input_gate_bias_ = {0.02234832, 0.14757581, 0.18176508, 0.10380666, in TEST_P()
1175 input_gate_bias_ = {0.03, 0.15, 0.22, 0.38}; in TEST_P()
1386 input_gate_bias_ = AddNullInput(); in LSTMIntegerOpModel()
[all …]
|
| D | optional_tensor_test.cc | 76 input_gate_bias_ = AddNullInput(); in LSTMOpModel()
78 input_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
156 PopulateTensor(input_gate_bias_, f); in SetInputGateBias()
210 int input_gate_bias_; member in tflite::__anon4dd1434f0111::LSTMOpModel
|
| D | unidirectional_sequence_lstm_test.cc | 83 input_gate_bias_ = AddNullInput(); in UnidirectionalLSTMOpModel()
85 input_gate_bias_ = AddInput(TensorType_FLOAT32); in UnidirectionalLSTMOpModel()
181 PopulateTensor(input_gate_bias_, f); in SetInputGateBias()
249 int input_gate_bias_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
368 std::vector<float> input_gate_bias_; member in tflite::__anond8582fe00111::BaseUnidirectionalLstmTest
462 input_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
547 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
609 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
676 lstm.SetInputGateBias(input_gate_bias_); in TEST_P()
739 lstm.SetInputGateBias(input_gate_bias_); in TEST_P()
[all …]
|
| D | lstm_eval_test.cc | 353 PackWeightToTensor(&input_gate_bias_tensor_, input_gate_bias_, in GetInputBias()
355 input_gate_bias_tensor_.data.i32 = input_gate_bias_.data(); in GetInputBias()
546 std::vector<int32_t> input_gate_bias_ = { member in tflite::__anon39d466b60111::QuantizedLstmParam
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3139 input_gate_bias_ = AddNullInput(); in LSTMOpModel()
3141 input_gate_bias_ = AddInput(TensorType_FLOAT32); in LSTMOpModel()
3239 PopulateTensor(input_gate_bias_, f); in SetInputGateBias()
3306 int input_gate_bias_; member in tflite::__anonde5f765e0111::LSTMOpModel
3350 std::vector<float> input_gate_bias_; member in tflite::__anonde5f765e0111::BaseLstmTest
3418 input_gate_bias_ = {0., 0., 0., 0.}; in SetUp()
3501 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
3570 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
3790 input_gate_bias_ = {0.02234832, 0.14757581, 0.18176508, 0.10380666, in SetUp()
4335 lstm.SetInputGateBias(input_gate_bias_); in TEST_F()
[all …]
|