Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc327 void ConvertLSTMCellSimpleToFusedLSTM::SetCellLayerNormCoefficients() { in SetCellLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
545 SetCellLayerNormCoefficients() { in SetCellLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test_util.h210 void SetCellLayerNormCoefficients(std::vector<float> f) { in SetCellLayerNormCoefficients() function
H A Dlstm_test.cc199 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMOpModel
1520 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMIntegerOpModel
H A Dunidirectional_sequence_lstm_test.cc2702 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anon47ec8c2f0111::UnidirectionalSequenceLSTMIntegerOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3347 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anon1a74836d0111::LSTMOpModel