Home
last modified time | relevance | path

Searched refs:forget_layer_norm_coefficients (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h111 const TfLiteTensor* forget_layer_norm_coefficients,
147 const TfLiteTensor* forget_layer_norm_coefficients,
185 const TfLiteTensor* forget_layer_norm_coefficients,
211 const TfLiteTensor* forget_layer_norm_coefficients,
Dunidirectional_sequence_lstm.cc135 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
237 layer_norm_forget_scale = forget_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
626 const TfLiteTensor* forget_layer_norm_coefficients; in CheckInputTensorDimensions() local
630 &forget_layer_norm_coefficients)); in CheckInputTensorDimensions()
631 TF_LITE_ENSURE_EQ(context, forget_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
632 TF_LITE_ENSURE_EQ(context, forget_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
635 TF_LITE_ENSURE_TYPES_EQ(context, forget_layer_norm_coefficients->type, in CheckInputTensorDimensions()
638 TF_LITE_ENSURE_TYPES_EQ(context, forget_layer_norm_coefficients->type, in CheckInputTensorDimensions()
855 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in Prepare() local
857 if (forget_layer_norm_coefficients == nullptr) { in Prepare()
[all …]
Dlstm.cc220 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
322 layer_norm_forget_scale = forget_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
508 const TfLiteTensor* forget_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
542 const bool is_layer_norm_lstm = (forget_layer_norm_coefficients != nullptr); in PopulateQuantizedLstmParams8x8_8()
634 layer_norm_forget_weight_ptr = forget_layer_norm_coefficients->data.i16; in PopulateQuantizedLstmParams8x8_8()
635 layer_norm_forget_scale = forget_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_8()
1078 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1080 TF_LITE_ENSURE(context, forget_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
1081 TF_LITE_ENSURE_EQ(context, forget_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
1082 TF_LITE_ENSURE_EQ(context, forget_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
[all …]
Dlstm_eval.cc1700 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat() argument
1787 GetTensorData<float>(forget_layer_norm_coefficients), in EvalFloat()
1848 GetTensorData<float>(forget_layer_norm_coefficients), in EvalFloat()
1889 const TfLiteTensor* forget_layer_norm_coefficients, in EvalHybrid() argument
2018 GetTensorData<float>(forget_layer_norm_coefficients), in EvalHybrid()
2116 GetTensorData<float>(forget_layer_norm_coefficients), in EvalHybrid()
2163 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_16() argument
2248 GetTensorData<int16_t>(forget_layer_norm_coefficients), in EvalInteger8x8_16()
2347 GetTensorData<int16_t>(forget_layer_norm_coefficients), in EvalInteger8x8_16()
2401 const TfLiteTensor* forget_layer_norm_coefficients, in EvalInteger8x8_8() argument
[all …]
Dlstm_test.cc1641 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
1721 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
1809 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
1894 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
1976 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
2061 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
Dunidirectional_sequence_lstm_test.cc3077 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
3157 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
3241 const std::vector<float> forget_layer_norm_coefficients = {0.2, 0.2, 0.4, in TEST() local
3326 lstm.SetForgetLayerNormCoefficients(forget_layer_norm_coefficients); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc276 const TfLiteTensor* forget_layer_norm_coefficients, in EvalCalibration() argument
365 GetTensorData<float>(forget_layer_norm_coefficients), in EvalCalibration()
427 GetTensorData<float>(forget_layer_norm_coefficients), in EvalCalibration()
523 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
613 input_layer_norm_coefficients, forget_layer_norm_coefficients, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc344 const TfLiteTensor* forget_layer_norm_coefficients, in EvalFloat() argument
434 GetTensorData<float>(forget_layer_norm_coefficients), in EvalFloat()
496 GetTensorData<float>(forget_layer_norm_coefficients), in EvalFloat()
567 const TfLiteTensor* forget_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
617 input_layer_norm_coefficients, forget_layer_norm_coefficients, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc58 std::initializer_list<float> forget_layer_norm_coefficients, in LSTMOpModel() argument
137 forget_layer_norm_coefficients); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc163 lstm.forget_layer_norm_coefficients(), lstm.forget_gate_bias(), builder); in LoadForLSTMOp()
Dprepare_quantize_helper.h103 !op.forget_layer_norm_coefficients().getType().template isa<NoneType>(); in GetLstmProperty()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc2046 !op.forget_layer_norm_coefficients().getType().isa<NoneType>(); in Verify()
2048 const auto forget_layer_norm_coefficients = in Verify() local
2049 op.forget_layer_norm_coefficients().getType().cast<ShapedType>(); in Verify()
2052 if (!forget_layer_norm_coefficients.hasRank() || in Verify()
2053 forget_layer_norm_coefficients.getRank() != 1 || in Verify()
2054 forget_layer_norm_coefficients.getDimSize(0) != n_cell) in Verify()
Dtfl_ops.td3870 TFL_TensorOfOrNone<[F32, QI16]>:$forget_layer_norm_coefficients,
3994 TFL_TensorOfOrNone<[F32, QI8]>:$forget_layer_norm_coefficients,