Home
last modified time | relevance | path

Searched refs:output_layer_norm_coefficients_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc220 std::initializer_list<float> output_layer_norm_coefficients_; member in tflite::__anonb664ae370111::LstmOpTest
333 output_layer_norm_coefficients_); in TEST_P()
401 output_layer_norm_coefficients_); in TEST_P()
964 output_layer_norm_coefficients_); in TEST_P()
1020 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in TEST_F()
1050 output_layer_norm_coefficients_); in TEST_F()
1087 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in TEST_F()
1113 output_layer_norm_coefficients_); in TEST_F()
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc120 output_layer_norm_coefficients_ = in LSTMOpModel()
126 output_layer_norm_coefficients_ = AddNullInput(); in LSTMOpModel()
204 PopulateTensor(output_layer_norm_coefficients_, f); in SetOutputLayerNormCoefficients()
261 int output_layer_norm_coefficients_ = kTfLiteOptionalTensor; member in tflite::__anonfa79752f0111::LSTMOpModel
333 std::vector<float> output_layer_norm_coefficients_; member in tflite::__anonfa79752f0111::LstmOpTest
408 lstm->SetOutputLayerNormCoefficients(output_layer_norm_coefficients_); in SetAllWeightsAndBiases()
1204 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in TEST_P()
1280 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in TEST_P()
1436 output_layer_norm_coefficients_ = AddInput( in LSTMIntegerOpModel()
1525 QuantizeAndPopulate<int16_t>(output_layer_norm_coefficients_, f); in SetOutputLayerNormCoefficients()
[all …]
Dunidirectional_sequence_lstm_test.cc121 output_layer_norm_coefficients_ = in UnidirectionalLSTMOpModel()
217 PopulateTensor(output_layer_norm_coefficients_, f); in SetOutputLayerNormCoefficients()
263 int output_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalLSTMOpModel
2504 std::vector<float> output_layer_norm_coefficients_; member in tflite::__anond8582fe00111::BaseLayerNormUnidirectionalLstmTest
2595 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in SetUp()
2670 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients_); in TEST_F()
2868 output_layer_norm_coefficients_ = AddInput( in UnidirectionalSequenceLSTMIntegerOpModel()
2959 QuantizeAndPopulate<int16_t>(output_layer_norm_coefficients_, f); in SetOutputLayerNormCoefficients()
3014 int output_layer_norm_coefficients_; member in tflite::__anond8582fe00111::UnidirectionalSequenceLSTMIntegerOpModel
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.h153 Value output_layer_norm_coefficients_; variable
Dlstm_utils.cc336 output_layer_norm_coefficients_ = none_; in SetOutputLayerNormCoefficients()
418 output_layer_norm_coefficients_, builder_.getStringAttr("TANH"), in RewriteFunc()
572 output_layer_norm_coefficients_ = in SetOutputLayerNormCoefficients()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3181 output_layer_norm_coefficients_ = in LSTMOpModel()
3275 PopulateTensor(output_layer_norm_coefficients_, f); in SetOutputLayerNormCoefficients()
3319 int output_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::LSTMOpModel
3365 std::vector<float> output_layer_norm_coefficients_; member in tflite::__anonde5f765e0111::BaseLstmTest
4402 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in SetUp()
4494 output_layer_norm_coefficients_); in TEST_F()
4544 output_layer_norm_coefficients_ = {0.6, 0.2, 0.2, 0.5}; in SetUp()
4630 output_layer_norm_coefficients_); in TEST_F()