Home
last modified time | relevance | path

Searched refs:SetInputLayerNormCoefficients (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h99 virtual void SetInputLayerNormCoefficients();
196 void SetInputLayerNormCoefficients() override;
Dlstm_utils.cc328 void ConvertLSTMCellSimpleToFusedLSTM::SetInputLayerNormCoefficients() { in SetInputLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
375 SetInputLayerNormCoefficients(); in GenerateFusedOpOperands()
549 SetInputLayerNormCoefficients() { in SetInputLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc191 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMOpModel
405 lstm->SetInputLayerNormCoefficients(input_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1512 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMIntegerOpModel
1720 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
1893 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
Dunidirectional_sequence_lstm_test.cc204 void SetInputLayerNormCoefficients(std::vector<float> f) { in SetInputLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2946 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
3156 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
3325 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3262 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anonde5f765e0111::LSTMOpModel
4489 layer_norm_lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients_); in TEST_F()