| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 218 std::initializer_list<float> forget_layer_norm_coefficients_; member in tflite::__anonb664ae370111::LstmOpTest
332 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_P()
400 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_P()
963 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_P()
1018 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in TEST_F()
1049 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_F()
1085 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in TEST_F()
1112 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 116 forget_layer_norm_coefficients_ = in LSTMOpModel()
124 forget_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
196 PopulateTensor(forget_layer_norm_coefficients_, f); in SetForgetLayerNormCoefficients()
259 int forget_layer_norm_coefficients_ = kTfLiteOptionalTensor; member in tflite::__anonfa79752f0111::LSTMOpModel
331 std::vector<float> forget_layer_norm_coefficients_; member in tflite::__anonfa79752f0111::LstmOpTest
406 lstm->SetForgetLayerNormCoefficients(forget_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1202 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in TEST_P()
1278 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in TEST_P()
1432 forget_layer_norm_coefficients_ = AddInput( in LSTMIntegerOpModel()
1517 QuantizeAndPopulate<int16_t>(forget_layer_norm_coefficients_, f); in SetForgetLayerNormCoefficients()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 117 forget_layer_norm_coefficients_ = in UnidirectionalLSTMOpModel()
209 PopulateTensor(forget_layer_norm_coefficients_, f); in SetForgetLayerNormCoefficients()
261 int forget_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2502 std::vector<float> forget_layer_norm_coefficients_; member in tflite::__anond8582fe00111::BaseLayerNormUnidirectionalLstmTest
2593 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in SetUp()
2668 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients_); in TEST_F()
2864 forget_layer_norm_coefficients_ = AddInput( in UnidirectionalSequenceLSTMIntegerOpModel()
2951 QuantizeAndPopulate<int16_t>(forget_layer_norm_coefficients_, f); in SetForgetLayerNormCoefficients()
3012 int forget_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.h | 151 Value forget_layer_norm_coefficients_; variable
|
| D | lstm_utils.cc | 333 forget_layer_norm_coefficients_ = none_; in SetForgetLayerNormCoefficients()
417 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in RewriteFunc()
563 forget_layer_norm_coefficients_ = in SetForgetLayerNormCoefficients()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3177 forget_layer_norm_coefficients_ = in LSTMOpModel()
3267 PopulateTensor(forget_layer_norm_coefficients_, f); in SetForgetLayerNormCoefficients()
3317 int forget_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::LSTMOpModel
3363 std::vector<float> forget_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::BaseLstmTest
4400 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in SetUp()
4491 forget_layer_norm_coefficients_); in TEST_F()
4542 forget_layer_norm_coefficients_ = {0.2, 0.2, 0.4, 0.3}; in SetUp()
4627 forget_layer_norm_coefficients_); in TEST_F()
|