| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 219 std::initializer_list<float> cell_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()
1019 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in TEST_F()
1049 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_F()
1086 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in TEST_F()
1112 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 118 cell_layer_norm_coefficients_ = in LSTMOpModel()
125 cell_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
200 PopulateTensor(cell_layer_norm_coefficients_, f); in SetCellLayerNormCoefficients()
260 int cell_layer_norm_coefficients_ = kTfLiteOptionalTensor; member in tflite::__anonfa79752f0111::LSTMOpModel
332 std::vector<float> cell_layer_norm_coefficients_; member in tflite::__anonfa79752f0111::LstmOpTest
407 lstm->SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1203 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in TEST_P()
1279 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in TEST_P()
1434 cell_layer_norm_coefficients_ = AddInput( in LSTMIntegerOpModel()
1521 QuantizeAndPopulate<int16_t>(cell_layer_norm_coefficients_, f); in SetCellLayerNormCoefficients()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 119 cell_layer_norm_coefficients_ = in UnidirectionalLSTMOpModel()
213 PopulateTensor(cell_layer_norm_coefficients_, f); in SetCellLayerNormCoefficients()
262 int cell_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2503 std::vector<float> cell_layer_norm_coefficients_; member in tflite::__anond8582fe00111::BaseLayerNormUnidirectionalLstmTest
2594 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in SetUp()
2669 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()
2866 cell_layer_norm_coefficients_ = AddInput( in UnidirectionalSequenceLSTMIntegerOpModel()
2955 QuantizeAndPopulate<int16_t>(cell_layer_norm_coefficients_, f); in SetCellLayerNormCoefficients()
3013 int cell_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.h | 152 Value cell_layer_norm_coefficients_; variable
|
| D | lstm_utils.cc | 325 cell_layer_norm_coefficients_ = none_; in SetCellLayerNormCoefficients()
417 forget_layer_norm_coefficients_, cell_layer_norm_coefficients_, in RewriteFunc()
542 cell_layer_norm_coefficients_ = in SetCellLayerNormCoefficients()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3179 cell_layer_norm_coefficients_ = in LSTMOpModel()
3271 PopulateTensor(cell_layer_norm_coefficients_, f); in SetCellLayerNormCoefficients()
3318 int cell_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::LSTMOpModel
3364 std::vector<float> cell_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::BaseLstmTest
4401 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in SetUp()
4492 layer_norm_lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()
4543 cell_layer_norm_coefficients_ = {0.7, 0.2, 0.3, 0.8}; in SetUp()
4628 layer_norm_lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()
|