Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
H A Dlstm_utils.cc331 void ConvertLSTMCellSimpleToFusedLSTM::SetInputLayerNormCoefficients() { in SetInputLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
554 SetInputLayerNormCoefficients() { in SetInputLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dunidirectional_sequence_lstm_test_util.h202 void SetInputLayerNormCoefficients(std::vector<float> f) { in SetInputLayerNormCoefficients() function
H A Dlstm_test.cc191 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMOpModel
1512 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anon299f347e0111::LSTMIntegerOpModel
H A Dunidirectional_sequence_lstm_test.cc2694 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anon47ec8c2f0111::UnidirectionalSequenceLSTMIntegerOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc3339 void SetInputLayerNormCoefficients(const std::vector<float>& f) { in SetInputLayerNormCoefficients() function in tflite::__anon1a74836d0111::LSTMOpModel