| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 110 const TfLiteTensor* input_layer_norm_coefficients,
146 const TfLiteTensor* input_layer_norm_coefficients,
184 const TfLiteTensor* input_layer_norm_coefficients,
210 const TfLiteTensor* input_layer_norm_coefficients,
|
| D | unidirectional_sequence_lstm.cc | 133 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
235 layer_norm_input_scale = input_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
608 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
611 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients, nullptr); in CheckInputTensorDimensions()
613 TF_LITE_ENSURE(context, input_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
614 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
615 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
618 TF_LITE_ENSURE_TYPES_EQ(context, input_layer_norm_coefficients->type, in CheckInputTensorDimensions()
621 TF_LITE_ENSURE_TYPES_EQ(context, input_layer_norm_coefficients->type, in CheckInputTensorDimensions()
1285 const TfLiteTensor* input_layer_norm_coefficients = in Eval() local
[all …]
|
| D | lstm.cc | 218 const TfLiteTensor* input_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
320 layer_norm_input_scale = input_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
506 const TfLiteTensor* input_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
631 layer_norm_input_weight_ptr = input_layer_norm_coefficients->data.i16; in PopulateQuantizedLstmParams8x8_8()
632 layer_norm_input_scale = input_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_8()
1060 const TfLiteTensor* input_layer_norm_coefficients = GetOptionalInputTensor( in CheckInputTensorDimensions() local
1063 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients, nullptr); in CheckInputTensorDimensions()
1065 TF_LITE_ENSURE(context, input_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
1066 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
1067 TF_LITE_ENSURE_EQ(context, input_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
[all …]
|
| D | lstm_eval.cc | 1699 const TfLiteTensor* input_layer_norm_coefficients, in EvalFloat() argument
1786 GetTensorData<float>(input_layer_norm_coefficients), in EvalFloat()
1847 GetTensorData<float>(input_layer_norm_coefficients), in EvalFloat()
1888 const TfLiteTensor* input_layer_norm_coefficients, in EvalHybrid() argument
2017 GetTensorData<float>(input_layer_norm_coefficients), in EvalHybrid()
2115 GetTensorData<float>(input_layer_norm_coefficients), in EvalHybrid()
2162 const TfLiteTensor* input_layer_norm_coefficients, in EvalInteger8x8_16() argument
2245 GetTensorData<int16_t>(input_layer_norm_coefficients), in EvalInteger8x8_16()
2344 GetTensorData<int16_t>(input_layer_norm_coefficients), in EvalInteger8x8_16()
2400 const TfLiteTensor* input_layer_norm_coefficients, in EvalInteger8x8_8() argument
[all …]
|
| D | lstm_test.cc | 1640 const std::vector<float> input_layer_norm_coefficients = {0.1, 0.2, 0.3, 0.5}; in TEST() local
1720 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
1808 const std::vector<float> input_layer_norm_coefficients = {0.1, 0.2, 0.3, 0.5}; in TEST() local
1893 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
1975 const std::vector<float> input_layer_norm_coefficients = {0.1, 0.2, 0.3, 0.5}; in TEST() local
|
| D | unidirectional_sequence_lstm_test.cc | 3076 const std::vector<float> input_layer_norm_coefficients = {0.1, 0.2, 0.3, 0.5}; in TEST() local
3156 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
3240 const std::vector<float> input_layer_norm_coefficients = {0.1, 0.2, 0.3, 0.5}; in TEST() local
3325 lstm.SetInputLayerNormCoefficients(input_layer_norm_coefficients); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 275 const TfLiteTensor* input_layer_norm_coefficients, in EvalCalibration() argument
364 GetTensorData<float>(input_layer_norm_coefficients), in EvalCalibration()
426 GetTensorData<float>(input_layer_norm_coefficients), in EvalCalibration()
520 const TfLiteTensor* input_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/ |
| D | lstm.cc | 343 const TfLiteTensor* input_layer_norm_coefficients, in EvalFloat() argument
433 GetTensorData<float>(input_layer_norm_coefficients), in EvalFloat()
495 GetTensorData<float>(input_layer_norm_coefficients), in EvalFloat()
564 const TfLiteTensor* input_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/ |
| D | lstm_full_test.cc | 57 std::initializer_list<float> input_layer_norm_coefficients, in LSTMOpModel() argument
134 input_layer_norm_coefficients); in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 156 lstm.input_layer_norm_coefficients(), lstm.input_gate_bias(), builder); in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3734 TypeIsPred<"input_layer_norm_coefficients", NoneType>]>,
3869 TFL_TensorOfOrNone<[F32, QI16]>:$input_layer_norm_coefficients,
3993 TFL_TensorOfOrNone<[F32, QI8]>:$input_layer_norm_coefficients,
|