Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dlstm_eval.h101 const TfLiteTensor* input_to_cell_weights,
131 const TfLiteTensor* input_to_cell_weights,
175 const TfLiteTensor* input_to_cell_weights,
201 const TfLiteTensor* input_to_cell_weights,
Dlstm.cc187 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
190 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_16()
333 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
475 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_8() local
478 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_8()
653 input_to_cell_weight_ptr = input_to_cell_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
654 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
886 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
889 &input_to_cell_weights)); in CheckInputTensorDimensions()
890 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
[all …]
Dunidirectional_sequence_lstm.cc98 const TfLiteTensor* input_to_cell_weights; in PopulateQuantizedLstmParams8x8_16() local
101 &input_to_cell_weights)); in PopulateQuantizedLstmParams8x8_16()
248 input_to_cell_weight_scale = input_to_cell_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
438 const TfLiteTensor* input_to_cell_weights; in CheckInputTensorDimensions() local
441 &input_to_cell_weights)); in CheckInputTensorDimensions()
442 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckInputTensorDimensions()
443 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckInputTensorDimensions()
444 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckInputTensorDimensions()
725 const TfLiteTensor* input_to_cell_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
728 &input_to_cell_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
Dlstm_eval.cc1690 const TfLiteTensor* input_to_cell_weights, in EvalFloat() argument
1773 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
1834 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
1873 const TfLiteTensor* input_to_cell_weights, in EvalHybrid() argument
1985 GetTensorData<int8_t>(input_to_cell_weights), in EvalHybrid()
1987 GetTensorScale(input_to_cell_weights), in EvalHybrid()
2083 GetTensorData<int8_t>(input_to_cell_weights), in EvalHybrid()
2085 GetTensorScale(input_to_cell_weights), in EvalHybrid()
2153 const TfLiteTensor* input_to_cell_weights, in EvalInteger8x8_16() argument
2212 GetTensorData<int8_t>(input_to_cell_weights), in EvalInteger8x8_16()
[all …]
Dlstm_test.cc1612 const std::vector<float> input_to_cell_weights = { in TEST() local
1704 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
1774 const std::vector<float> input_to_cell_weights = { in TEST() local
1873 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
1947 const std::vector<float> input_to_cell_weights = { in TEST() local
2043 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
2137 const std::vector<float>& input_to_cell_weights, in HybridSparseLSTMOpModel() argument
2162 AddConstSparseInput(input_weights_td, input_to_cell_weights, true); in HybridSparseLSTMOpModel()
Dbidirectional_sequence_lstm.cc217 const TfLiteTensor* input_to_cell_weights; in CheckLstmTensorDimensionsAndTypes() local
220 &input_to_cell_weights)); in CheckLstmTensorDimensionsAndTypes()
221 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->size, 2); in CheckLstmTensorDimensionsAndTypes()
222 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[0], n_cell); in CheckLstmTensorDimensionsAndTypes()
223 TF_LITE_ENSURE_EQ(context, input_to_cell_weights->dims->data[1], n_input); in CheckLstmTensorDimensionsAndTypes()
224 TF_LITE_ENSURE_TYPES_EQ(context, input_to_cell_weights->type, in CheckLstmTensorDimensionsAndTypes()
Dunidirectional_sequence_lstm_test.cc3048 const std::vector<float> input_to_cell_weights = { in TEST() local
3140 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
3206 const std::vector<float> input_to_cell_weights = { in TEST() local
3305 lstm.SetInputToCellWeights(input_to_cell_weights); in TEST()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/
Dlstm.cc266 const TfLiteTensor* input_to_cell_weights, in EvalCalibration() argument
351 GetTensorData<float>(input_to_cell_weights), in EvalCalibration()
413 GetTensorData<float>(input_to_cell_weights), in EvalCalibration()
480 const TfLiteTensor* input_to_cell_weights; in lstm_eval() local
484 &input_to_cell_weights)); in lstm_eval()
609 input_to_cell_weights, input_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/
Dlstm.cc334 const TfLiteTensor* input_to_cell_weights, in EvalFloat() argument
420 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
482 GetTensorData<float>(input_to_cell_weights), in EvalFloat()
543 const TfLiteTensor* input_to_cell_weights = GetInput( in lstm_eval() local
613 input_to_cell_weights, input_to_output_weights, in lstm_eval()
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc42 std::initializer_list<float> input_to_cell_weights, in LSTMOpModel() argument
73 AddConstInput({weight_type, {n_cell, n_input}}, input_to_cell_weights); in LSTMOpModel()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dload_quantization_recipe.cc165 auto cell_gate = CreateGate(loc, lstm.input(), lstm.input_to_cell_weights(), in LoadForLSTMOp()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td3797 TFL_OperandHasRank<3, 2>, // input_to_cell_weights
3837 TFL_TensorOf<[F32, QI8]>:$input_to_cell_weights,
3926 TFL_OperandHasRank<3, 2>, // input_to_cell_weights
3961 TFL_TensorOf<[F32, QI8]>:$input_to_cell_weights,