Home
last modified time | relevance | path

Searched refs:output_layer_norm_coefficients (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h113 const TfLiteTensor* output_layer_norm_coefficients,
149 const TfLiteTensor* output_layer_norm_coefficients,
187 const TfLiteTensor* output_layer_norm_coefficients,
213 const TfLiteTensor* output_layer_norm_coefficients,
Dunidirectional_sequence_lstm.cc139 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
239 layer_norm_output_scale = output_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
658 const TfLiteTensor* output_layer_norm_coefficients; in CheckInputTensorDimensions() local
662 &output_layer_norm_coefficients)); in CheckInputTensorDimensions()
663 TF_LITE_ENSURE_EQ(context, output_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
664 TF_LITE_ENSURE_EQ(context, output_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
667 TF_LITE_ENSURE_TYPES_EQ(context, output_layer_norm_coefficients->type, in CheckInputTensorDimensions()
670 TF_LITE_ENSURE_TYPES_EQ(context, output_layer_norm_coefficients->type, in CheckInputTensorDimensions()
1298 const TfLiteTensor* output_layer_norm_coefficients = in Eval() local
1327 cell_layer_norm_coefficients, output_layer_norm_coefficients, in Eval()
[all …]
Dlstm.cc224 const TfLiteTensor* output_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
324 layer_norm_output_scale = output_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
512 const TfLiteTensor* output_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
638 layer_norm_output_weight_ptr = output_layer_norm_coefficients->data.i16; in PopulateQuantizedLstmParams8x8_8()
639 layer_norm_output_scale = output_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_8()
1106 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1108 TF_LITE_ENSURE(context, output_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
1109 TF_LITE_ENSURE_EQ(context, output_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
1110 TF_LITE_ENSURE_EQ(context, output_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
1113 TF_LITE_ENSURE_TYPES_EQ(context, output_layer_norm_coefficients->type, in CheckInputTensorDimensions()
[all …]
Dlstm_eval.cc1702 const TfLiteTensor* output_layer_norm_coefficients, in EvalFloat() argument
1789 GetTensorData<float>(output_layer_norm_coefficients), in EvalFloat()
1850 GetTensorData<float>(output_layer_norm_coefficients), in EvalFloat()
1891 const TfLiteTensor* output_layer_norm_coefficients, in EvalHybrid() argument
2020 GetTensorData<float>(output_layer_norm_coefficients), in EvalHybrid()
2118 GetTensorData<float>(output_layer_norm_coefficients), in EvalHybrid()
2165 const TfLiteTensor* output_layer_norm_coefficients, in EvalInteger8x8_16() argument
2254 GetTensorData<int16_t>(output_layer_norm_coefficients), in EvalInteger8x8_16()
2353 GetTensorData<int16_t>(output_layer_norm_coefficients), in EvalInteger8x8_16()
2403 const TfLiteTensor* output_layer_norm_coefficients, in EvalInteger8x8_8() argument
[all …]
Dlstm_test.cc1644 const std::vector<float> output_layer_norm_coefficients = {0.6, 0.2, 0.2, in TEST() local
1723 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
1812 const std::vector<float> output_layer_norm_coefficients = {0.6, 0.2, 0.2, in TEST() local
1896 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
1979 const std::vector<float> output_layer_norm_coefficients = {0.6, 0.2, 0.2, in TEST() local
2063 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
Dunidirectional_sequence_lstm_test.cc3080 const std::vector<float> output_layer_norm_coefficients = {0.6, 0.2, 0.2, in TEST() local
3159 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
3244 const std::vector<float> output_layer_norm_coefficients = {0.6, 0.2, 0.2, in TEST() local
3328 lstm.SetOutputLayerNormCoefficients(output_layer_norm_coefficients); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc346 const TfLiteTensor* output_layer_norm_coefficients, in EvalFloat() argument
436 GetTensorData<float>(output_layer_norm_coefficients), in EvalFloat()
498 GetTensorData<float>(output_layer_norm_coefficients), in EvalFloat()
573 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
618 cell_layer_norm_coefficients, output_layer_norm_coefficients, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc278 const TfLiteTensor* output_layer_norm_coefficients, in EvalCalibration() argument
367 GetTensorData<float>(output_layer_norm_coefficients), in EvalCalibration()
429 GetTensorData<float>(output_layer_norm_coefficients), in EvalCalibration()
529 const TfLiteTensor* output_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
614 cell_layer_norm_coefficients, output_layer_norm_coefficients, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc60 std::initializer_list<float> output_layer_norm_coefficients) in LSTMOpModel()
141 output_layer_norm_coefficients); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc183 lstm.output_layer_norm_coefficients(), lstm.output_gate_bias(), builder); in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3872 TFL_TensorOfOrNone<[F32, QI16]>:$output_layer_norm_coefficients,
3996 TFL_TensorOfOrNone<[F32, QI8]>:$output_layer_norm_coefficients,