This commit is contained in:
jomjol
2021-09-12 07:29:30 +02:00
parent dd995ec28a
commit 7fcb5d1c0c
18 changed files with 104 additions and 32 deletions

View File

@@ -194,6 +194,8 @@ void CTfLiteClass::MakeAllocate()
TfLiteStatus allocate_status = this->interpreter->AllocateTensors();
if (allocate_status != kTfLiteOk) {
TF_LITE_REPORT_ERROR(error_reporter, "AllocateTensors() failed");
LogFile.WriteToFile("AllocateTensors() failed");
this->GetInputDimension();
return;
}