Searched refs:SetForgetLayerNormCoefficients (Results 1 – 5 of 5) sorted by relevance
100 virtual void SetForgetLayerNormCoefficients();197 void SetForgetLayerNormCoefficients() override;
332 void ConvertLSTMCellSimpleToFusedLSTM::SetForgetLayerNormCoefficients() { in SetForgetLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM376 SetForgetLayerNormCoefficients(); in GenerateFusedOpOperands()561 SetForgetLayerNormCoefficients() { in SetForgetLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
195 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMOpModel406 lstm->SetForgetLayerNormCoefficients(forget_layer_norm_coefficients_); in SetAllWeightsAndBiases()1516 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMIntegerOpModel1721 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()1894 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()2061 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
208 void SetForgetLayerNormCoefficients(std::vector<float> f) { in SetForgetLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel2668 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients_); in TEST_F()2950 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel3157 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()3326 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
3266 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anonde5f765e0111::LSTMOpModel4490 layer_norm_lstm.SetForgetLayerNormCoefficients( in TEST_F()4626 layer_norm_lstm.SetForgetLayerNormCoefficients( in TEST_F()