Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm.cc183 const TfLiteTensor* input_to_forget_weights; in PopulateQuantizedLstmParams8x8_16() local
186 &input_to_forget_weights)); in PopulateQuantizedLstmParams8x8_16()
332 input_to_forget_weight_scale = input_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
471 const TfLiteTensor* input_to_forget_weights; in PopulateQuantizedLstmParams8x8_8() local
474 &input_to_forget_weights)); in PopulateQuantizedLstmParams8x8_8()
651 input_to_forget_weight_ptr = input_to_forget_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
652 input_to_forget_weight_scale = input_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
864 const TfLiteTensor* input_to_forget_weights; in CheckInputTensorDimensions() local
867 &input_to_forget_weights)); in CheckInputTensorDimensions()
868 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dlstm_eval.h100 const TfLiteTensor* input_to_forget_weights,
129 const TfLiteTensor* input_to_forget_weights,
174 const TfLiteTensor* input_to_forget_weights,
200 const TfLiteTensor* input_to_forget_weights,
Dbidirectional_sequence_lstm.cc196 const TfLiteTensor* input_to_forget_weights; in CheckLstmTensorDimensionsAndTypes() local
199 &input_to_forget_weights)); in CheckLstmTensorDimensionsAndTypes()
200 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
201 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
202 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
203 TF_LITE_ENSURE(context, (input_to_forget_weights->type == kTfLiteFloat32) || in CheckLstmTensorDimensionsAndTypes()
204 (input_to_forget_weights->type == kTfLiteInt8) || in CheckLstmTensorDimensionsAndTypes()
205 (input_to_forget_weights->type == kTfLiteUInt8)); in CheckLstmTensorDimensionsAndTypes()
214 input_to_forget_weights->type); in CheckLstmTensorDimensionsAndTypes()
225 input_to_forget_weights->type); in CheckLstmTensorDimensionsAndTypes()
[all …]
Dunidirectional_sequence_lstm.cc93 const TfLiteTensor* input_to_forget_weights; in PopulateQuantizedLstmParams8x8_16() local
97 &input_to_forget_weights)); in PopulateQuantizedLstmParams8x8_16()
247 input_to_forget_weight_scale = input_to_forget_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
429 const TfLiteTensor* input_to_forget_weights; in CheckInputTensorDimensions() local
433 &input_to_forget_weights)); in CheckInputTensorDimensions()
434 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->size, 2); in CheckInputTensorDimensions()
435 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
436 TF_LITE_ENSURE_EQ(context, input_to_forget_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
493 is_integer ? kTfLiteInt16 : input_to_forget_weights->type); in CheckInputTensorDimensions()
503 is_integer ? kTfLiteInt16 : input_to_forget_weights->type); in CheckInputTensorDimensions()
[all …]
Dlstm_eval.cc1689 const TfLiteTensor* input_to_forget_weights, in EvalFloat() argument
1772 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
1833 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
1871 const TfLiteTensor* input_to_forget_weights, in EvalHybrid() argument
1982 GetTensorData<int8_t>(input_to_forget_weights), in EvalHybrid()
1984 GetTensorScale(input_to_forget_weights), in EvalHybrid()
2080 GetTensorData<int8_t>(input_to_forget_weights), in EvalHybrid()
2082 GetTensorScale(input_to_forget_weights), in EvalHybrid()
2152 const TfLiteTensor* input_to_forget_weights, in EvalInteger8x8_16() argument
2209 GetTensorData<int8_t>(input_to_forget_weights), in EvalInteger8x8_16()
[all …]
Dlstm_test.cc1608 const std::vector<float> input_to_forget_weights = { in TEST() local
1705 lstm.SetInputToForgetWeights(input_to_forget_weights); in TEST()
1770 const std::vector<float> input_to_forget_weights = { in TEST() local
1874 lstm.SetInputToForgetWeights(input_to_forget_weights); in TEST()
1943 const std::vector<float> input_to_forget_weights = { in TEST() local
2044 lstm.SetInputToForgetWeights(input_to_forget_weights); in TEST()
2136 const std::vector<float>& input_to_forget_weights, in HybridSparseLSTMOpModel() argument
2159 AddConstSparseInput(input_weights_td, input_to_forget_weights, true); in HybridSparseLSTMOpModel()
Dunidirectional_sequence_lstm_test.cc3044 const std::vector<float> input_to_forget_weights = { in TEST() local
3141 lstm.SetInputToForgetWeights(input_to_forget_weights); in TEST()
3202 const std::vector<float> input_to_forget_weights = { in TEST() local
3306 lstm.SetInputToForgetWeights(input_to_forget_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc265 const TfLiteTensor* input_to_forget_weights, in EvalCalibration() argument
350 GetTensorData<float>(input_to_forget_weights), in EvalCalibration()
412 GetTensorData<float>(input_to_forget_weights), in EvalCalibration()
474 const TfLiteTensor* input_to_forget_weights; in lstm_eval() local
479 &input_to_forget_weights)); in lstm_eval()
608 input, input_to_input_weights, input_to_forget_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc333 const TfLiteTensor* input_to_forget_weights, in EvalFloat() argument
419 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
481 GetTensorData<float>(input_to_forget_weights), in EvalFloat()
541 const TfLiteTensor* input_to_forget_weights = GetInput( in lstm_eval() local
612 input, input_to_input_weights, input_to_forget_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc41 std::initializer_list<float> input_to_forget_weights, in LSTMOpModel() argument
72 AddConstInput({weight_type, {n_cell, n_input}}, input_to_forget_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc159 loc, lstm.input(), lstm.input_to_forget_weights(), in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3796 TFL_OperandHasRank<2, 2>, // input_to_forget_weights
3836 TFL_TensorOf<[F32, QI8]>:$input_to_forget_weights,
3925 TFL_OperandHasRank<2, 2>, // input_to_forget_weights
3960 TFL_TensorOf<[F32, QI8]>:$input_to_forget_weights,