experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

lstm_eval_tests.cc: Give zero_points_tensorN_ a proper name.

Browse Source 
PiperOrigin-RevId: 318569888
Change-Id: I21d55643be277ef84d3ee56ca4a50462f1e5aaea
This commit is contained in:
Robert David 2020-06-26 17:04:22 -07:00 committed by TensorFlower Gardener
parent cd463b2bab
commit 432b1c1c43
1 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
tensorflow/lite/kernels/lstm_eval_test.cc
View File

@ -695,22 +695,22 @@ class HybridLstmParam : public BaseLstmParam {
return &cell_quantized_tensor_; return &cell_quantized_tensor_;
} }
TfLiteTensor* GetInputZeroPoints() { TfLiteTensor* GetInputZeroPoints() {
PackWeightToTensor(&zero_points_tensor0_, input_zp_, PackWeightToTensor(&input_zp_tensor_, input_zp_,
quantization_extra_scratch_buffer_sizes_); quantization_extra_scratch_buffer_sizes_);
zero_points_tensor0_.data.i32 = input_zp_.data(); input_zp_tensor_.data.i32 = input_zp_.data();
return &zero_points_tensor0_; return &input_zp_tensor_;
} }
TfLiteTensor* GetAuxInputZeroPoints() { TfLiteTensor* GetAuxInputZeroPoints() {
PackWeightToTensor(&zero_points_tensor1_, aux_input_zp_, PackWeightToTensor(&aux_input_zp_tensor_, aux_input_zp_,
quantization_extra_scratch_buffer_sizes_); quantization_extra_scratch_buffer_sizes_);
zero_points_tensor1_.data.i32 = aux_input_zp_.data(); aux_input_zp_tensor_.data.i32 = aux_input_zp_.data();
return &zero_points_tensor1_; return &aux_input_zp_tensor_;
} }
TfLiteTensor* GetOutputStateZeroPoints() { TfLiteTensor* GetOutputStateZeroPoints() {
PackWeightToTensor(&zero_points_tensor2_, output_state_zp_, PackWeightToTensor(&output_state_zp_tensor_, output_state_zp_,
quantization_extra_scratch_buffer_sizes_); quantization_extra_scratch_buffer_sizes_);
zero_points_tensor2_.data.i32 = output_state_zp_.data(); output_state_zp_tensor_.data.i32 = output_state_zp_.data();
return &zero_points_tensor2_; return &output_state_zp_tensor_;
} }
TfLiteTensor* GetRowSums() { TfLiteTensor* GetRowSums() {
PackWeightToTensor(&row_sums_tensor_, row_sums_, row_sums_size_); PackWeightToTensor(&row_sums_tensor_, row_sums_, row_sums_size_);
@ -808,9 +808,9 @@ class HybridLstmParam : public BaseLstmParam {
TfLiteIntArrayFree(input_quantized_tensor_.dims); TfLiteIntArrayFree(input_quantized_tensor_.dims);
TfLiteIntArrayFree(activation_quantized_tensor_.dims); TfLiteIntArrayFree(activation_quantized_tensor_.dims);
TfLiteIntArrayFree(cell_quantized_tensor_.dims); TfLiteIntArrayFree(cell_quantized_tensor_.dims);
TfLiteIntArrayFree(zero_points_tensor0_.dims); TfLiteIntArrayFree(input_zp_tensor_.dims);
TfLiteIntArrayFree(zero_points_tensor1_.dims); TfLiteIntArrayFree(aux_input_zp_tensor_.dims);
TfLiteIntArrayFree(zero_points_tensor2_.dims); TfLiteIntArrayFree(output_state_zp_tensor_.dims);
TfLiteIntArrayFree(row_sums_tensor_.dims); TfLiteIntArrayFree(row_sums_tensor_.dims);
} }
@ -833,11 +833,11 @@ class HybridLstmParam : public BaseLstmParam {
TfLiteTensor prod_scaling_factors_tensor_; TfLiteTensor prod_scaling_factors_tensor_;
std::vector<int32_t> input_zp_; std::vector<int32_t> input_zp_;
TfLiteTensor zero_points_tensor0_; TfLiteTensor input_zp_tensor_;
std::vector<int32_t> aux_input_zp_; std::vector<int32_t> aux_input_zp_;
TfLiteTensor zero_points_tensor1_; TfLiteTensor aux_input_zp_tensor_;
std::vector<int32_t> output_state_zp_; std::vector<int32_t> output_state_zp_;
TfLiteTensor zero_points_tensor2_; TfLiteTensor output_state_zp_tensor_;
std::vector<int8_t> input_quantized_; std::vector<int8_t> input_quantized_;
TfLiteTensor input_quantized_tensor_; TfLiteTensor input_quantized_tensor_;