OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fixed a bug where slice Op legalization constructing QNN param with slicing size instead of end index.

Browse Source 
PiperOrigin-RevId: 713450306
This commit is contained in:
Andrew Zhang
2025-01-08 16:12:13 -08:00
committed by TensorFlower Gardener
parent 7886562dd6
commit aa290d76bb
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/experimental/litert/vendors/qualcomm/compiler/legalizations/slice_op_legalization.cc vendored
View File

@@ -110,7 +110,8 @@ LiteRtStatus SliceOpLegalization::LegalizeOp(const Op& src,
// Copy begin, end, and stride values from src_begin_indices and
// src_size_indices to range_tensor_data. Stride is always 1.
range_tensor_data[i * kRangesParamArgSize] = src_begin_indices->at(i);
range_tensor_data[i * kRangesParamArgSize + 1] = src_size_indices->at(i);
range_tensor_data[i * kRangesParamArgSize + 1] =
src_begin_indices->at(i) + src_size_indices->at(i);
range_tensor_data[i * kRangesParamArgSize + 2] = 1;
}