Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h98 virtual void SetCellLayerNormCoefficients();
195 void SetCellLayerNormCoefficients() override;
Dlstm_utils.cc324 void ConvertLSTMCellSimpleToFusedLSTM::SetCellLayerNormCoefficients() { in SetCellLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
374 SetCellLayerNormCoefficients(); in GenerateFusedOpOperands()
540 SetCellLayerNormCoefficients() { in SetCellLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc199 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMOpModel
407 lstm->SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1520 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMIntegerOpModel
1722 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
1895 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
2062 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
Dunidirectional_sequence_lstm_test.cc212 void SetCellLayerNormCoefficients(std::vector<float> f) { in SetCellLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2669 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()
2954 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
3158 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
3327 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3270 void SetCellLayerNormCoefficients(const std::vector<float>& f) { in SetCellLayerNormCoefficients() function in tflite::__anonde5f765e0111::LSTMOpModel
4492 layer_norm_lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()
4628 layer_norm_lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients_); in TEST_F()