| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 108 const TfLiteTensor* cell_to_forget_weights,
144 const TfLiteTensor* cell_to_forget_weights,
182 const TfLiteTensor* cell_to_forget_weights,
208 const TfLiteTensor* cell_to_forget_weights,
|
| D | unidirectional_sequence_lstm.cc | 128 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
229 cell_to_forget_weight_scale = cell_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
496 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
498 if (cell_to_forget_weights != nullptr) { in CheckInputTensorDimensions()
499 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->size, 1); in CheckInputTensorDimensions()
500 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
502 context, cell_to_forget_weights->type, in CheckInputTensorDimensions()
520 (cell_to_forget_weights != nullptr) && in CheckInputTensorDimensions()
523 (cell_to_forget_weights == nullptr) && in CheckInputTensorDimensions()
1253 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in Eval() local
[all …]
|
| D | lstm.cc | 213 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_16() local
314 cell_to_forget_weight_scale = cell_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
501 const TfLiteTensor* cell_to_forget_weights = in PopulateQuantizedLstmParams8x8_8() local
623 cell_to_forget_weight_ptr = cell_to_forget_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
625 cell_to_forget_weight_scale = cell_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
950 const TfLiteTensor* cell_to_forget_weights = in CheckInputTensorDimensions() local
952 if (cell_to_forget_weights) { in CheckInputTensorDimensions()
953 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->size, 1); in CheckInputTensorDimensions()
954 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
956 context, cell_to_forget_weights->type, in CheckInputTensorDimensions()
[all …]
|
| D | lstm_eval.cc | 1697 const TfLiteTensor* cell_to_forget_weights, in EvalFloat() argument
1784 GetTensorData<float>(cell_to_forget_weights), in EvalFloat()
1845 GetTensorData<float>(cell_to_forget_weights), in EvalFloat()
1886 const TfLiteTensor* cell_to_forget_weights, in EvalHybrid() argument
2013 GetTensorData<int8_t>(cell_to_forget_weights), in EvalHybrid()
2014 GetTensorScale(cell_to_forget_weights), in EvalHybrid()
2111 GetTensorData<int8_t>(cell_to_forget_weights), in EvalHybrid()
2112 GetTensorScale(cell_to_forget_weights), in EvalHybrid()
2160 const TfLiteTensor* cell_to_forget_weights, in EvalInteger8x8_16() argument
2233 GetTensorData<int16_t>(cell_to_forget_weights), in EvalInteger8x8_16()
[all …]
|
| D | bidirectional_sequence_lstm.cc | 290 const TfLiteTensor* cell_to_forget_weights = in CheckLstmTensorDimensionsAndTypes() local
292 if (cell_to_forget_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
293 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->size, 1); in CheckLstmTensorDimensionsAndTypes()
294 TF_LITE_ENSURE_EQ(context, cell_to_forget_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
295 TF_LITE_ENSURE_TYPES_EQ(context, cell_to_forget_weights->type, in CheckLstmTensorDimensionsAndTypes()
312 (cell_to_forget_weights != nullptr) && in CheckLstmTensorDimensionsAndTypes()
315 (cell_to_forget_weights == nullptr) && in CheckLstmTensorDimensionsAndTypes()
|
| D | lstm_test.cc | 1804 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
1888 lstm.SetCellToForgetWeights(cell_to_forget_weights); in TEST()
|
| D | unidirectional_sequence_lstm_test.cc | 3236 const std::vector<float> cell_to_forget_weights = {0.2, -0.1, 0.1, -0.2}; in TEST() local
3320 lstm.SetCellToForgetWeights(cell_to_forget_weights); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 273 const TfLiteTensor* cell_to_forget_weights, in EvalCalibration() argument
362 GetTensorData<float>(cell_to_forget_weights), in EvalCalibration()
424 GetTensorData<float>(cell_to_forget_weights), in EvalCalibration()
515 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
612 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 341 const TfLiteTensor* cell_to_forget_weights, in EvalFloat() argument
431 GetTensorData<float>(cell_to_forget_weights), in EvalFloat()
493 GetTensorData<float>(cell_to_forget_weights), in EvalFloat()
559 const TfLiteTensor* cell_to_forget_weights = GetOptionalInputTensor( in lstm_eval() local
616 cell_to_input_weights, cell_to_forget_weights, cell_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 49 std::initializer_list<float> cell_to_forget_weights, in LSTMOpModel() argument
94 AddConstInput({weight_type, {n_cell}}, cell_to_forget_weights); in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 162 {lstm.input_cell_state(), lstm.cell_to_forget_weights()}), in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 3802 TFL_OperandIsNoneOrHasRank<10, 1>, // cell_to_forget_weights
3849 TFL_TensorOfOrNone<[F32, QI8, QI16]>:$cell_to_forget_weights,
3933 TFL_OperandIsNoneOrHasRank<10, 1>, // cell_to_forget_weights
3973 TFL_TensorOfOrNone<[F32, QI8]>:$cell_to_forget_weights,
|