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

Fix array access in Obj-C TFLInterpreter::resizeInputTensorAtIndex

Browse Source 
PiperOrigin-RevId: 335955632
Change-Id: Ie7192d3cf2d926e0dcb24430b4abb6e5c4c3f960
This commit is contained in:
A. Unique TensorFlower 2020-10-07 14:38:11 -07:00 committed by TensorFlower Gardener
parent 43208d2c8c
commit e9947ca30a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/experimental/objc/sources/TFLInterpreter.mm
View File

@ -201,7 +201,7 @@ static void TFLInterpreterErrorReporter(void *user_data, const char *format, va_
return NO;
}
std::vector<int> cDimensions(self.inputTensorCount);
std::vector<int> cDimensions(shape.count);
for (int dimIndex = 0; dimIndex < shape.count; ++dimIndex) {
int dimension = shape[dimIndex].intValue;
if (dimension <= 0) {