| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 217 std::initializer_list<float> input_layer_norm_coefficients_; member in tflite::__anonb664ae370111::LstmOpTest
331 projection_weights_, {}, input_layer_norm_coefficients_, in TEST_P()
399 projection_weights_, {}, input_layer_norm_coefficients_, in TEST_P()
962 projection_weights_, {}, input_layer_norm_coefficients_, in TEST_P()
1017 input_layer_norm_coefficients_ = {0.1, 0.2, 0.3, 0.5}; in TEST_F()
1048 projection_weights_, {}, input_layer_norm_coefficients_, in TEST_F()
1111 projection_weights_, {}, input_layer_norm_coefficients_, in TEST_F()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_test.cc | 111 input_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
113 input_layer_norm_coefficients_ = in LSTMOpModel()
123 input_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
192 PopulateTensor(input_layer_norm_coefficients_, f); in SetInputLayerNormCoefficients()
258 int input_layer_norm_coefficients_ = kTfLiteOptionalTensor; member in tflite::__anonfa79752f0111::LSTMOpModel
330 std::vector<float> input_layer_norm_coefficients_; member in tflite::__anonfa79752f0111::LstmOpTest
405 lstm->SetInputLayerNormCoefficients(input_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1201 input_layer_norm_coefficients_ = {0.1, 0.2, 0.3, 0.5}; in TEST_P()
1427 input_layer_norm_coefficients_ = AddNullInput(); in LSTMIntegerOpModel()
1429 input_layer_norm_coefficients_ = AddInput( in LSTMIntegerOpModel()
[all …]
|
| D | unidirectional_sequence_lstm_test.cc | 112 input_layer_norm_coefficients_ = AddNullInput(); in UnidirectionalLSTMOpModel()
114 input_layer_norm_coefficients_ = in UnidirectionalLSTMOpModel()
205 PopulateTensor(input_layer_norm_coefficients_, f); in SetInputLayerNormCoefficients()
260 int input_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2501 std::vector<float> input_layer_norm_coefficients_; member in tflite::__anond8582fe00111::BaseLayerNormUnidirectionalLstmTest
2592 input_layer_norm_coefficients_ = {0.1, 0.2, 0.3, 0.5}; in SetUp()
2859 input_layer_norm_coefficients_ = AddNullInput(); in UnidirectionalSequenceLSTMIntegerOpModel()
2861 input_layer_norm_coefficients_ = AddInput( in UnidirectionalSequenceLSTMIntegerOpModel()
2947 QuantizeAndPopulate<int16_t>(input_layer_norm_coefficients_, f); in SetInputLayerNormCoefficients()
3011 int input_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.h | 150 Value input_layer_norm_coefficients_; variable
|
| D | lstm_utils.cc | 329 input_layer_norm_coefficients_ = none_; in SetInputLayerNormCoefficients()
416 input_cell_state_, input_layer_norm_coefficients_, in RewriteFunc()
551 input_layer_norm_coefficients_ = in SetInputLayerNormCoefficients()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3172 input_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
3174 input_layer_norm_coefficients_ = in LSTMOpModel()
3263 PopulateTensor(input_layer_norm_coefficients_, f); in SetInputLayerNormCoefficients()
3316 int input_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::LSTMOpModel
3362 std::vector<float> input_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::BaseLstmTest
4399 input_layer_norm_coefficients_ = {0.1, 0.2, 0.3, 0.5}; in SetUp()
4489 layer_norm_lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients_); in TEST_F()
|