| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 112 const TfLiteTensor* cell_layer_norm_coefficients,
148 const TfLiteTensor* cell_layer_norm_coefficients,
186 const TfLiteTensor* cell_layer_norm_coefficients,
212 const TfLiteTensor* cell_layer_norm_coefficients,
|
| D | unidirectional_sequence_lstm.cc | 137 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
238 layer_norm_cell_scale = cell_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
642 const TfLiteTensor* cell_layer_norm_coefficients; in CheckInputTensorDimensions() local
646 &cell_layer_norm_coefficients)); in CheckInputTensorDimensions()
647 TF_LITE_ENSURE_EQ(context, cell_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
648 TF_LITE_ENSURE_EQ(context, cell_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
651 TF_LITE_ENSURE_TYPES_EQ(context, cell_layer_norm_coefficients->type, in CheckInputTensorDimensions()
654 TF_LITE_ENSURE_TYPES_EQ(context, cell_layer_norm_coefficients->type, in CheckInputTensorDimensions()
1294 const TfLiteTensor* cell_layer_norm_coefficients = in Eval() local
1327 cell_layer_norm_coefficients, output_layer_norm_coefficients, in Eval()
[all …]
|
| D | lstm.cc | 222 const TfLiteTensor* cell_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_16() local
323 layer_norm_cell_scale = cell_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_16()
510 const TfLiteTensor* cell_layer_norm_coefficients = in PopulateQuantizedLstmParams8x8_8() local
636 layer_norm_cell_weight_ptr = cell_layer_norm_coefficients->data.i16; in PopulateQuantizedLstmParams8x8_8()
637 layer_norm_cell_scale = cell_layer_norm_coefficients->params.scale; in PopulateQuantizedLstmParams8x8_8()
1092 const TfLiteTensor* cell_layer_norm_coefficients = in CheckInputTensorDimensions() local
1094 TF_LITE_ENSURE(context, cell_layer_norm_coefficients != nullptr); in CheckInputTensorDimensions()
1095 TF_LITE_ENSURE_EQ(context, cell_layer_norm_coefficients->dims->size, 1); in CheckInputTensorDimensions()
1096 TF_LITE_ENSURE_EQ(context, cell_layer_norm_coefficients->dims->data[0], in CheckInputTensorDimensions()
1099 TF_LITE_ENSURE_TYPES_EQ(context, cell_layer_norm_coefficients->type, in CheckInputTensorDimensions()
[all …]
|
| D | lstm_eval.cc | 1701 const TfLiteTensor* cell_layer_norm_coefficients, in EvalFloat() argument
1788 GetTensorData<float>(cell_layer_norm_coefficients), in EvalFloat()
1849 GetTensorData<float>(cell_layer_norm_coefficients), in EvalFloat()
1890 const TfLiteTensor* cell_layer_norm_coefficients, in EvalHybrid() argument
2019 GetTensorData<float>(cell_layer_norm_coefficients), in EvalHybrid()
2117 GetTensorData<float>(cell_layer_norm_coefficients), in EvalHybrid()
2164 const TfLiteTensor* cell_layer_norm_coefficients, in EvalInteger8x8_16() argument
2251 GetTensorData<int16_t>(cell_layer_norm_coefficients), in EvalInteger8x8_16()
2350 GetTensorData<int16_t>(cell_layer_norm_coefficients), in EvalInteger8x8_16()
2402 const TfLiteTensor* cell_layer_norm_coefficients, in EvalInteger8x8_8() argument
[all …]
|
| D | lstm_test.cc | 1643 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
1722 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
1811 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
1895 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
1978 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
2062 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
|
| D | unidirectional_sequence_lstm_test.cc | 3079 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
3158 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
3243 const std::vector<float> cell_layer_norm_coefficients = {0.7, 0.2, 0.3, 0.8}; in TEST() local
3327 lstm.SetCellLayerNormCoefficients(cell_layer_norm_coefficients); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 277 const TfLiteTensor* cell_layer_norm_coefficients, in EvalCalibration() argument
366 GetTensorData<float>(cell_layer_norm_coefficients), in EvalCalibration()
428 GetTensorData<float>(cell_layer_norm_coefficients), in EvalCalibration()
526 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
614 cell_layer_norm_coefficients, output_layer_norm_coefficients, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 345 const TfLiteTensor* cell_layer_norm_coefficients, in EvalFloat() argument
435 GetTensorData<float>(cell_layer_norm_coefficients), in EvalFloat()
497 GetTensorData<float>(cell_layer_norm_coefficients), in EvalFloat()
570 const TfLiteTensor* cell_layer_norm_coefficients = GetOptionalInputTensor( in lstm_eval() local
618 cell_layer_norm_coefficients, output_layer_norm_coefficients, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 59 std::initializer_list<float> cell_layer_norm_coefficients, in LSTMOpModel()
139 cell_layer_norm_coefficients); in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 168 lstm.cell_layer_norm_coefficients(), in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3871 TFL_TensorOfOrNone<[F32, QI16]>:$cell_layer_norm_coefficients,
3995 TFL_TensorOfOrNone<[F32, QI8]>:$cell_layer_norm_coefficients,
|