Home
last modified time | relevance | path

Searched refs:use_layer_norm (Results 1 – 19 of 19) sorted by relevance

/external/ComputeLibrary/tests/validation/fixtures/
DLSTMLayerFixture.h52 bool use_layer_norm) in setup() argument
55 data_type, projection_opt, peephole_opt, use_layer_norm); in setup()
57 data_type, projection_opt, peephole_opt, use_layer_norm); in setup()
81 …jection_threshold, DataType data_type, bool projection_opt, bool peephole_opt, bool use_layer_norm) in compute_target() argument
146 if(use_layer_norm) in compute_target()
288 if(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() argument
376 if(use_layer_norm) in compute_reference()
395 if(!cifg_opt && use_layer_norm) in compute_reference()
420 if(use_layer_norm) in compute_reference()
[all …]
/external/tensorflow/tensorflow/lite/tools/optimize/
Doperator_property.cc45 op_variant.use_layer_norm = false; in GetOperatorVariant()
47 op_variant.use_layer_norm = in GetOperatorVariant()
290 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
361 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
427 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
498 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
563 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
633 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()
697 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
766 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
Doperator_property.h136 bool use_layer_norm = false; member
/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_lstm.cc41 bool use_layer_norm; member
90 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
164 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()
241 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()
420 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions() argument
615 if (use_layer_norm) { in CheckInputTensorDimensions()
781 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()
863 bool use_layer_norm = false; in Prepare() local
868 use_layer_norm = false; in Prepare()
870 use_layer_norm = true; in Prepare()
[all …]
Dlstm.cc52 bool use_layer_norm; member
177 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local
245 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()
319 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()
856 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions() argument
1060 if (use_layer_norm) { in CheckInputTensorDimensions()
1213 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()
1300 op_data->use_layer_norm = false; in Prepare()
1302 op_data->use_layer_norm = true; in Prepare()
1306 op_data->use_layer_norm = false; in Prepare()
[all …]
Dlstm_eval.cc193 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
197 if (use_layer_norm) { in CalculateLstmGateFloat()
229 if (use_layer_norm) { in CalculateLstmGateFloat()
378 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateHybrid() local
382 if (use_layer_norm) { in CalculateLstmGateHybrid()
445 if (use_layer_norm) { in CalculateLstmGateHybrid()
557 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateInteger8x8_16() local
581 if (use_layer_norm) { in CalculateLstmGateInteger8x8_16()
Dlstm_test.cc1317 bool use_layer_norm, bool use_8x8_8_implementation, in LSTMIntegerOpModel() argument
1425 if (use_layer_norm) { in LSTMIntegerOpModel()
Dunidirectional_sequence_lstm_test.cc2496 bool use_layer_norm, bool use_8x8_8_implementation, in UnidirectionalSequenceLSTMIntegerOpModel() argument
2605 if (use_layer_norm) { in UnidirectionalSequenceLSTMIntegerOpModel()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayer.cpp79 _is_layer_norm_lstm = lstm_params.use_layer_norm(); in configure()
422 if(lstm_params.use_layer_norm()) in validate()
473 …onnectedLayer::validate(input, input_to_forget_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
480 if(lstm_params.use_layer_norm()) in validate()
505 …::validate(input, lstm_params.input_to_input_weights(), (lstm_params.use_layer_norm()) ? nullptr :… in validate()
515 if(lstm_params.use_layer_norm()) in validate()
529 …yConnectedLayer::validate(input, input_to_cell_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
532 if(lstm_params.use_layer_norm()) in validate()
557 …onnectedLayer::validate(input, input_to_output_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
565 if(lstm_params.use_layer_norm()) in validate()
DNEQLSTMLayer.cpp289 _has_layer_norm = lstm_params.use_layer_norm(); in configure()
828 const bool has_layer_norm = lstm_params.use_layer_norm(); in validate()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc74 const bool use_layer_norm = (forget_layer_norm_coefficients_ptr != nullptr); in LstmStepWithAuxInput() local
78 if (use_layer_norm) { in LstmStepWithAuxInput()
209 if (use_layer_norm) { in LstmStepWithAuxInput()
231 if (use_layer_norm) { in LstmStepWithAuxInput()
249 if (use_layer_norm) { in LstmStepWithAuxInput()
282 if (use_layer_norm) { in LstmStepWithAuxInput()
534 bool use_layer_norm; member
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayer.cpp92 _is_layer_norm_lstm = lstm_params.use_layer_norm(); in configure()
447 if(lstm_params.use_layer_norm()) in validate()
491 …onnectedLayer::validate(input, input_to_forget_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
506 if(lstm_params.use_layer_norm()) in validate()
532 …::validate(input, lstm_params.input_to_input_weights(), (lstm_params.use_layer_norm()) ? nullptr :… in validate()
542 if(lstm_params.use_layer_norm()) in validate()
556 …yConnectedLayer::validate(input, input_to_cell_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
559 if(lstm_params.use_layer_norm()) in validate()
583 …onnectedLayer::validate(input, input_to_output_weights, (lstm_params.use_layer_norm()) ? nullptr :… in validate()
591 if(lstm_params.use_layer_norm()) in validate()
DCLQLSTMLayer.cpp218 _has_layer_norm = lstm_params.use_layer_norm(); in configure()
737 const bool has_layer_norm = lstm_params.use_layer_norm(); in validate()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc53 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local
57 if (use_layer_norm) { in CalculateLstmGateFloat()
84 if (use_layer_norm) { in CalculateLstmGateFloat()
456 bool use_layer_norm; member
/external/ComputeLibrary/arm_compute/core/utils/misc/
DInfoHelpers.h93 if(lstm_params.use_layer_norm()) in build_lstm_params_tensor_info()
/external/ComputeLibrary/arm_compute/runtime/common/
DLSTMParams.h310 bool use_layer_norm() const in use_layer_norm() function
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h103 lstm_variant->use_layer_norm = in GetLstmProperty()
435 if (!lstm_variant.use_layer_norm && index != 4) { in processIntermediates()
/external/ComputeLibrary/utils/
DTypePrinter.h3143 << "use_layer_norm=" << lstm_params.use_layer_norm() << "}";
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc3240 const bool use_layer_norm = input_shapes.size() > 20; in LSTMOpModel() local
3242 if (use_layer_norm) { in LSTMOpModel()