Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h103 const TfLiteTensor* recurrent_to_input_weights,
135 const TfLiteTensor* recurrent_to_input_weights,
177 const TfLiteTensor* recurrent_to_input_weights,
203 const TfLiteTensor* recurrent_to_input_weights,
Dlstm.cc196 const TfLiteTensor* recurrent_to_input_weights = in PopulateQuantizedLstmParams8x8_16() local
307 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
484 const TfLiteTensor* recurrent_to_input_weights = in PopulateQuantizedLstmParams8x8_8() local
612 recurrent_to_input_weight_ptr = recurrent_to_input_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
615 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
896 const TfLiteTensor* recurrent_to_input_weights = in CheckInputTensorDimensions() local
898 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
899 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
900 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
902 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
[all …]
Dunidirectional_sequence_lstm.cc108 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulateQuantizedLstmParams8x8_16() local
222 recurrent_to_input_weight_scale = recurrent_to_input_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
446 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in CheckInputTensorDimensions() local
448 if (recurrent_to_input_weights != nullptr) { in CheckInputTensorDimensions()
449 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckInputTensorDimensions()
450 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckInputTensorDimensions()
452 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckInputTensorDimensions()
481 (recurrent_to_input_weights != nullptr)) || in CheckInputTensorDimensions()
483 (recurrent_to_input_weights == nullptr)); in CheckInputTensorDimensions()
735 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in PopulatePrecomputedZPTimesWeightsWithBias() local
[all …]
Dlstm_eval.cc1692 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat() argument
1779 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
1840 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
1877 const TfLiteTensor* recurrent_to_input_weights, in EvalHybrid() argument
1999 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalHybrid()
2001 GetTensorScale(recurrent_to_input_weights), in EvalHybrid()
2097 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalHybrid()
2099 GetTensorScale(recurrent_to_input_weights), in EvalHybrid()
2155 const TfLiteTensor* recurrent_to_input_weights, in EvalInteger8x8_16() argument
2218 GetTensorData<int8_t>(recurrent_to_input_weights), in EvalInteger8x8_16()
[all …]
Dbidirectional_sequence_lstm.cc237 const TfLiteTensor* recurrent_to_input_weights = in CheckLstmTensorDimensionsAndTypes() local
239 if (recurrent_to_input_weights != nullptr) { in CheckLstmTensorDimensionsAndTypes()
240 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
241 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[0], in CheckLstmTensorDimensionsAndTypes()
243 TF_LITE_ENSURE_EQ(context, recurrent_to_input_weights->dims->data[1], in CheckLstmTensorDimensionsAndTypes()
245 TF_LITE_ENSURE_TYPES_EQ(context, recurrent_to_input_weights->type, in CheckLstmTensorDimensionsAndTypes()
276 (recurrent_to_input_weights != nullptr)) || in CheckLstmTensorDimensionsAndTypes()
278 (recurrent_to_input_weights == nullptr)); in CheckLstmTensorDimensionsAndTypes()
Dlstm_test.cc1628 const std::vector<float> recurrent_to_input_weights = { in TEST() local
1713 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
1790 const std::vector<float> recurrent_to_input_weights = { in TEST() local
1882 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
1963 const std::vector<float> recurrent_to_input_weights = { in TEST() local
2140 const std::vector<float>& recurrent_to_input_weights, in HybridSparseLSTMOpModel() argument
2171 recurrent_weights_td, recurrent_to_input_weights, true); in HybridSparseLSTMOpModel()
Dunidirectional_sequence_lstm_test.cc3064 const std::vector<float> recurrent_to_input_weights = { in TEST() local
3149 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
3222 const std::vector<float> recurrent_to_input_weights = { in TEST() local
3314 lstm.SetRecurrentToInputWeights(recurrent_to_input_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc268 const TfLiteTensor* recurrent_to_input_weights, in EvalCalibration() argument
357 GetTensorData<float>(recurrent_to_input_weights), in EvalCalibration()
419 GetTensorData<float>(recurrent_to_input_weights), in EvalCalibration()
492 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( in lstm_eval() local
610 recurrent_to_input_weights, recurrent_to_forget_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc336 const TfLiteTensor* recurrent_to_input_weights, in EvalFloat() argument
426 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
488 GetTensorData<float>(recurrent_to_input_weights), in EvalFloat()
548 const TfLiteTensor* recurrent_to_input_weights = GetOptionalInputTensor( 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.cc44 std::initializer_list<float> recurrent_to_input_weights, in LSTMOpModel() argument
80 recurrent_to_input_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc153 lstm.input_activation_state(), lstm.recurrent_to_input_weights(), in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3731 TypeIsPred<"recurrent_to_input_weights", NoneType>,
3737 TypeIsPred<"recurrent_to_input_weights", NoneType>,
3798 TFL_OperandIsNoneOrHasRank<5, 2>, // recurrent_to_input_weights
3841 TFL_TensorOfOrNone<[F32, QI8]>:$recurrent_to_input_weights,
3928 TFL_OperandIsNoneOrHasRank<5, 2>, // recurrent_to_input_weights
3965 TFL_TensorOfOrNone<[F32, QI8]>:$recurrent_to_input_weights,
4244 TFL_TensorOf<[F32, QI8]>:$recurrent_to_input_weights,