Home
last modified time | relevance | path

Searched defs:SetForgetLayerNormCoefficients (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc335 void ConvertLSTMCellSimpleToFusedLSTM::SetForgetLayerNormCoefficients() { in SetForgetLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
566 SetForgetLayerNormCoefficients() { in SetForgetLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test_util.h206 void SetForgetLayerNormCoefficients(std::vector<float> f) { in SetForgetLayerNormCoefficients() function
H A Dlstm_test.cc195 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMOpModel
1516 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMIntegerOpModel
H A Dunidirectional_sequence_lstm_test.cc2698 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anon47ec8c2f0111::UnidirectionalSequenceLSTMIntegerOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3343 void SetForgetLayerNormCoefficients(const std::vector<float>& f) { in SetForgetLayerNormCoefficients() function in tflite::__anon1a74836d0111::LSTMOpModel