| /external/tensorflow/tensorflow/lite/kernels/ |
| D | lstm_eval.h | 106 const TfLiteTensor* recurrent_to_output_weights,
141 const TfLiteTensor* recurrent_to_output_weights,
180 const TfLiteTensor* recurrent_to_output_weights,
206 const TfLiteTensor* recurrent_to_output_weights,
|
| D | unidirectional_sequence_lstm.cc | 120 const TfLiteTensor* recurrent_to_output_weights; in PopulateQuantizedLstmParams8x8_16() local
124 &recurrent_to_output_weights)); in PopulateQuantizedLstmParams8x8_16()
252 recurrent_to_output_weight_scale = recurrent_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
747 const TfLiteTensor* recurrent_to_output_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
751 &recurrent_to_output_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
819 context, output_state_zero_point, recurrent_to_output_weights, in PopulatePrecomputedZPTimesWeightsWithBias()
897 const TfLiteTensor* recurrent_to_output_weights; in Prepare() local
901 &recurrent_to_output_weights)); in Prepare()
902 TF_LITE_ENSURE_EQ(context, recurrent_to_output_weights->dims->size, 2); in Prepare()
903 TF_LITE_ENSURE_EQ(context, recurrent_to_output_weights->dims->data[0], in Prepare()
[all …]
|
| D | lstm.cc | 206 const TfLiteTensor* recurrent_to_output_weights; in PopulateQuantizedLstmParams8x8_16() local
209 &recurrent_to_output_weights)); in PopulateQuantizedLstmParams8x8_16()
337 recurrent_to_output_weight_scale = recurrent_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_16()
494 const TfLiteTensor* recurrent_to_output_weights; in PopulateQuantizedLstmParams8x8_8() local
497 &recurrent_to_output_weights)); in PopulateQuantizedLstmParams8x8_8()
661 recurrent_to_output_weight_ptr = recurrent_to_output_weights->data.int8; in PopulateQuantizedLstmParams8x8_8()
662 recurrent_to_output_weight_scale = recurrent_to_output_weights->params.scale; in PopulateQuantizedLstmParams8x8_8()
1189 const TfLiteTensor* recurrent_to_output_weights; in PopulatePrecomputedZPTimesWeightsWithBias() local
1192 &recurrent_to_output_weights)); in PopulatePrecomputedZPTimesWeightsWithBias()
1255 context, output_state_zero_point, recurrent_to_output_weights, in PopulatePrecomputedZPTimesWeightsWithBias()
[all …]
|
| D | lstm_eval.cc | 1695 const TfLiteTensor* recurrent_to_output_weights, in EvalFloat() argument
1729 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalFloat()
1782 GetTensorData<float>(recurrent_to_output_weights), in EvalFloat()
1843 GetTensorData<float>(recurrent_to_output_weights), in EvalFloat()
1883 const TfLiteTensor* recurrent_to_output_weights, in EvalHybrid() argument
1927 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalHybrid()
2008 GetTensorData<int8_t>(recurrent_to_output_weights), in EvalHybrid()
2010 GetTensorScale(recurrent_to_output_weights), in EvalHybrid()
2106 GetTensorData<int8_t>(recurrent_to_output_weights), in EvalHybrid()
2108 GetTensorScale(recurrent_to_output_weights), in EvalHybrid()
[all …]
|
| D | lstm_test.cc | 1637 const std::vector<float> recurrent_to_output_weights = { in TEST() local
1716 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights); in TEST()
1799 const std::vector<float> recurrent_to_output_weights = { in TEST() local
1885 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights); in TEST()
1972 const std::vector<float> recurrent_to_output_weights = { in TEST() local
2055 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights); in TEST()
2143 const std::vector<float>& recurrent_to_output_weights, in HybridSparseLSTMOpModel() argument
2179 recurrent_weights_td, recurrent_to_output_weights, true); in HybridSparseLSTMOpModel()
|
| D | unidirectional_sequence_lstm_test.cc | 3073 const std::vector<float> recurrent_to_output_weights = { in TEST() local
3152 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights); in TEST()
3231 const std::vector<float> recurrent_to_output_weights = { in TEST() local
3317 lstm.SetRecurrentToOutputWeights(recurrent_to_output_weights); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
| D | lstm.cc | 271 const TfLiteTensor* recurrent_to_output_weights, in EvalCalibration() argument
307 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalCalibration()
360 GetTensorData<float>(recurrent_to_output_weights), in EvalCalibration()
422 GetTensorData<float>(recurrent_to_output_weights), in EvalCalibration()
506 const TfLiteTensor* recurrent_to_output_weights; in lstm_eval() local
511 &recurrent_to_output_weights)); in lstm_eval()
611 recurrent_to_cell_weights, recurrent_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
| D | lstm.cc | 339 const TfLiteTensor* recurrent_to_output_weights, in EvalFloat() argument
376 const int n_output = recurrent_to_output_weights->dims->data[1]; in EvalFloat()
429 GetTensorData<float>(recurrent_to_output_weights), in EvalFloat()
491 GetTensorData<float>(recurrent_to_output_weights), in EvalFloat()
554 const TfLiteTensor* recurrent_to_output_weights = GetInput( in lstm_eval() local
615 recurrent_to_cell_weights, recurrent_to_output_weights, in lstm_eval()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 47 std::initializer_list<float> recurrent_to_output_weights, in LSTMOpModel() argument
86 recurrent_to_output_weights); in LSTMOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 180 lstm.input_activation_state(), lstm.recurrent_to_output_weights(), in LoadForLSTMOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2022 const auto recurrent_to_output_weights = in Verify() local
2023 op.recurrent_to_output_weights().getType().cast<ShapedType>(); in Verify()
2026 recurrent_to_output_weights.hasStaticShape()) { in Verify()
2029 const int n_output = recurrent_to_output_weights.getDimSize(1); in Verify()
2039 (recurrent_to_output_weights.getRank() != 2) || in Verify()
2040 (recurrent_to_output_weights.getDimSize(0) != n_cell) || in Verify()
|
| D | tfl_ops.td | 3844 TFL_TensorOf<[F32, QI8]>:$recurrent_to_output_weights,
3931 TFL_OperandHasRank<8, 2>, // recurrent_to_output_weights
3968 TFL_TensorOf<[F32, QI8]>:$recurrent_to_output_weights,
|