Searched refs:use_layer_norm (Results 1 – 11 of 11) sorted by relevance
45 op_variant.use_layer_norm = false; in GetOperatorVariant()47 op_variant.use_layer_norm = in GetOperatorVariant()269 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()340 if (op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()406 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()477 if (op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()542 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()612 if (!op_variant.use_layer_norm && op_variant.use_projection && in GetOperatorProperty()676 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()745 if (!op_variant.use_layer_norm && !op_variant.use_projection && in GetOperatorProperty()
133 bool use_layer_norm = false; member
40 bool use_layer_norm; member85 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local159 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()233 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()412 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions() argument607 if (use_layer_norm) { in CheckInputTensorDimensions()771 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()853 bool use_layer_norm = false; in Prepare() local858 use_layer_norm = false; in Prepare()860 use_layer_norm = true; in Prepare()[all …]
52 bool use_layer_norm; member176 const bool use_layer_norm = op_data->use_layer_norm; in PopulateQuantizedLstmParams8x8_16() local244 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()318 if (use_layer_norm) { in PopulateQuantizedLstmParams8x8_16()855 bool use_layer_norm, bool is_integer) { in CheckInputTensorDimensions() argument1059 if (use_layer_norm) { in CheckInputTensorDimensions()1212 const bool is_layer_norm = op_data->use_layer_norm; in PopulatePrecomputedZPTimesWeightsWithBias()1299 op_data->use_layer_norm = false; in Prepare()1301 op_data->use_layer_norm = true; in Prepare()1305 op_data->use_layer_norm = false; in Prepare()[all …]
161 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local165 if (use_layer_norm) { in CalculateLstmGateFloat()192 if (use_layer_norm) { in CalculateLstmGateFloat()336 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateHybrid() local340 if (use_layer_norm) { in CalculateLstmGateHybrid()403 if (use_layer_norm) { in CalculateLstmGateHybrid()515 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateInteger8x8_16() local539 if (use_layer_norm) { in CalculateLstmGateInteger8x8_16()
1317 bool use_layer_norm, bool use_8x8_8_implementation, in LSTMIntegerOpModel() argument1425 if (use_layer_norm) { in LSTMIntegerOpModel()
2748 bool use_layer_norm, bool use_8x8_8_implementation, in UnidirectionalSequenceLSTMIntegerOpModel() argument2857 if (use_layer_norm) { in UnidirectionalSequenceLSTMIntegerOpModel()
74 const bool use_layer_norm = (forget_layer_norm_coefficients_ptr != nullptr); in LstmStepWithAuxInput() local78 if (use_layer_norm) { in LstmStepWithAuxInput()203 if (use_layer_norm) { in LstmStepWithAuxInput()224 if (use_layer_norm) { in LstmStepWithAuxInput()241 if (use_layer_norm) { in LstmStepWithAuxInput()274 if (use_layer_norm) { in LstmStepWithAuxInput()523 bool use_layer_norm; member
53 const bool use_layer_norm = (layer_norm_coefficients != nullptr); in CalculateLstmGateFloat() local57 if (use_layer_norm) { in CalculateLstmGateFloat()84 if (use_layer_norm) { in CalculateLstmGateFloat()455 bool use_layer_norm; member
102 lstm_variant->use_layer_norm = in GetLstmProperty()434 if (!lstm_variant.use_layer_norm && index != 4) { in processIntermediates()
3163 const bool use_layer_norm = input_shapes.size() > 20; in LSTMOpModel() local3165 if (use_layer_norm) { in LSTMOpModel()