Home
last modified time | relevance | path

Searched refs:recurrent_to_forget_weights (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h104 const TfLiteTensor* recurrent_to_forget_weights,
137 const TfLiteTensor* recurrent_to_forget_weights,
178 const TfLiteTensor* recurrent_to_forget_weights,
204 const TfLiteTensor* recurrent_to_forget_weights,
Dlstm.cc198 const TfLiteTensor* recurrent_to_forget_weights; in PopulateQuantizedLstmParams8x8_16() local
201 &recurrent_to_forget_weights)); in PopulateQuantizedLstmParams8x8_16()
335 recurrent_to_forget_weight_scale = recurrent_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
486 const TfLiteTensor* recurrent_to_forget_weights; in PopulateQuantizedLstmParams8x8_8() local
489 &recurrent_to_forget_weights)); in PopulateQuantizedLstmParams8x8_8()
657 recurrent_to_forget_weight_ptr = recurrent_to_forget_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
658 recurrent_to_forget_weight_scale = recurrent_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
908 const TfLiteTensor* recurrent_to_forget_weights; in CheckInputTensorDimensions() local
911 &recurrent_to_forget_weights)); in CheckInputTensorDimensions()
912 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dunidirectional_sequence_lstm.cc110 const TfLiteTensor* recurrent_to_forget_weights; in PopulateQuantizedLstmParams8x8_16() local
114 &recurrent_to_forget_weights)); in PopulateQuantizedLstmParams8x8_16()
250 recurrent_to_forget_weight_scale = recurrent_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
456 const TfLiteTensor* recurrent_to_forget_weights; in CheckInputTensorDimensions() local
460 &recurrent_to_forget_weights)); in CheckInputTensorDimensions()
461 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
462 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->data[0], in CheckInputTensorDimensions()
464 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->data[1], in CheckInputTensorDimensions()
737 const TfLiteTensor* recurrent_to_forget_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
741 &recurrent_to_forget_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Dlstm_eval.cc1693 const TfLiteTensor* recurrent_to_forget_weights, in EvalFloat() argument
1780 GetTensorData<float>(recurrent_to_forget_weights), in EvalFloat()
1841 GetTensorData<float>(recurrent_to_forget_weights), in EvalFloat()
1879 const TfLiteTensor* recurrent_to_forget_weights, in EvalHybrid() argument
2002 GetTensorData<int8_t>(recurrent_to_forget_weights), in EvalHybrid()
2004 GetTensorScale(recurrent_to_forget_weights), in EvalHybrid()
2100 GetTensorData<int8_t>(recurrent_to_forget_weights), in EvalHybrid()
2102 GetTensorScale(recurrent_to_forget_weights), in EvalHybrid()
2156 const TfLiteTensor* recurrent_to_forget_weights, in EvalInteger8x8_16() argument
2221 GetTensorData<int8_t>(recurrent_to_forget_weights), in EvalInteger8x8_16()
[all …]
Dlstm_test.cc1634 const std::vector<float> recurrent_to_forget_weights = { in TEST() local
1715 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights); in TEST()
1796 const std::vector<float> recurrent_to_forget_weights = { in TEST() local
1884 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights); in TEST()
1969 const std::vector<float> recurrent_to_forget_weights = { in TEST() local
2054 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights); in TEST()
2141 const std::vector<float>& recurrent_to_forget_weights, in HybridSparseLSTMOpModel() argument
2175 recurrent_weights_td, recurrent_to_forget_weights, true); in HybridSparseLSTMOpModel()
Dbidirectional_sequence_lstm.cc249 const TfLiteTensor* recurrent_to_forget_weights; in CheckLstmTensorDimensionsAndTypes() local
252 &recurrent_to_forget_weights)); in CheckLstmTensorDimensionsAndTypes()
253 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
254 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->data[0], in CheckLstmTensorDimensionsAndTypes()
256 TF_LITE_ENSURE_EQ(context, recurrent_to_forget_weights->dims->data[1], in CheckLstmTensorDimensionsAndTypes()
258 TF_LITE_ENSURE_TYPES_EQ(context, recurrent_to_forget_weights->type, in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc3070 const std::vector<float> recurrent_to_forget_weights = { in TEST() local
3151 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights); in TEST()
3228 const std::vector<float> recurrent_to_forget_weights = { in TEST() local
3316 lstm.SetRecurrentToForgetWeights(recurrent_to_forget_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc269 const TfLiteTensor* recurrent_to_forget_weights, in EvalCalibration() argument
358 GetTensorData<float>(recurrent_to_forget_weights), in EvalCalibration()
420 GetTensorData<float>(recurrent_to_forget_weights), in EvalCalibration()
494 const TfLiteTensor* recurrent_to_forget_weights; in lstm_eval() local
499 &recurrent_to_forget_weights)); in lstm_eval()
610 recurrent_to_input_weights, recurrent_to_forget_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc337 const TfLiteTensor* recurrent_to_forget_weights, in EvalFloat() argument
427 GetTensorData<float>(recurrent_to_forget_weights), in EvalFloat()
489 GetTensorData<float>(recurrent_to_forget_weights), in EvalFloat()
550 const TfLiteTensor* recurrent_to_forget_weights = GetInput( in lstm_eval() local
614 recurrent_to_input_weights, recurrent_to_forget_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc45 std::initializer_list<float> recurrent_to_forget_weights, in LSTMOpModel() argument
83 recurrent_to_forget_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc160 lstm.input_activation_state(), lstm.recurrent_to_forget_weights(), in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3799 TFL_OperandHasRank<6, 2>, // recurrent_to_forget_weights
3842 TFL_TensorOf<[F32, QI8]>:$recurrent_to_forget_weights,
3929 TFL_OperandHasRank<6, 2>, // recurrent_to_forget_weights
3966 TFL_TensorOf<[F32, QI8]>:$recurrent_to_forget_weights,