Searched defs:use_layer_norm (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | unidirectional_sequence_lstm.cc | 41 bool use_layer_norm; member
90 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
420 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions()
863 bool use_layer_norm = false; in Prepare() local
1223 const bool use_layer_norm = op_data->use_layer_norm; in Eval() local
|
| H A D | lstm.cc | 52 bool use_layer_norm; member
177 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
856 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions()
1314 const bool use_layer_norm = op_data->use_layer_norm; in Prepare() local
|
| H A D | lstm_eval.cc | 193 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
378 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateHybrid() local
557 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateInteger8x8_16() local
|
| H A D | unidirectional_sequence_lstm_test.cc | 2496 bool use_layer_norm, bool use_8x8_8_implementation, in UnidirectionalSequenceLSTMIntegerOpModel()
|
| H A D | lstm_test.cc | 1317 bool use_layer_norm, bool use_8x8_8_implementation, in LSTMIntegerOpModel()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | LSTMLayerFixture.h | 52 bool use_layer_norm) in setup()
81 …jection_threshold, DataType data_type, bool projection_opt, bool peephole_opt, bool use_layer_norm) in compute_target()
326 …jection_threshold, DataType data_type, bool projection_opt, bool peephole_opt, bool use_layer_norm) in compute_reference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| H A D | lstm.cc | 74 const bool use_layer_norm = (forget_layer_norm_coefficients_ptr != nullptr); in LstmStepWithAuxInput() local
534 bool use_layer_norm; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| H A D | lstm.cc | 53 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
456 bool use_layer_norm; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | operator_property.h | 136 bool use_layer_norm = false; member
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/common/ |
| H A D | LSTMParams.h | 310 bool use_layer_norm() const in use_layer_norm() function
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 37 bool use_layer_norm; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| H A D | nnapi_delegate_test.cc | 3240 const bool use_layer_norm = input_shapes.size() > 20; in LSTMOpModel() local
|