Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h101 virtual void SetOutputLayerNormCoefficients();
198 void SetOutputLayerNormCoefficients() override;
Dlstm_utils.cc335 void ConvertLSTMCellSimpleToFusedLSTM::SetOutputLayerNormCoefficients() { in SetOutputLayerNormCoefficients() function in mlir::TFL::ConvertLSTMCellSimpleToFusedLSTM
377 SetOutputLayerNormCoefficients(); in GenerateFusedOpOperands()
570 SetOutputLayerNormCoefficients() { in SetOutputLayerNormCoefficients() function in mlir::TFL::ConvertLayerNormalizedLSTMCellSimpleToFusedLSTM
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc203 void SetOutputLayerNormCoefficients(const std::vector<float>& f) { in SetOutputLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMOpModel
408 lstm->SetOutputLayerNormCoefficients(output_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1524 void SetOutputLayerNormCoefficients(const std::vector<float>& f) { in SetOutputLayerNormCoefficients() function in tflite::__anonfa79752f0111::LSTMIntegerOpModel
1723 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
1896 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
2063 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
Dunidirectional_sequence_lstm_test.cc216 void SetOutputLayerNormCoefficients(std::vector<float> f) { in SetOutputLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2670 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients_); in TEST_F()
2958 void SetOutputLayerNormCoefficients(const std::vector<float>& f) { in SetOutputLayerNormCoefficients() function in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
3159 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
3328 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3274 void SetOutputLayerNormCoefficients(const std::vector<float>& f) { in SetOutputLayerNormCoefficients() function in tflite::__anonde5f765e0111::LSTMOpModel
4493 layer_norm_lstm.SetOutputLayerNormCoefficients( in TEST_F()
4629 layer_norm_lstm.SetOutputLayerNormCoefficients( in TEST_F()