Merge pull request tensorflow#62998 from 2JS:master

PiperOrigin-RevId: 692966366
ROCm · Nov 4, 2024 · 8cf3f16 · 8cf3f16
2 parents 149bf5f + 037566e
commit 8cf3f16
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow/lite/swift/Sources/Interpreter.swift b/tensorflow/lite/swift/Sources/Interpreter.swift
@@ -66,6 +66,9 @@ public final class Interpreter {
   /// The underlying `TfLiteInterpreter` C pointer.
   internal var cInterpreter: CInterpreter?
 
+  /// Keep reference to underlying model's data in case of init(modelData:) is used.
+  internal let _model: Model
+
   /// The underlying `TfLiteDelegate` C pointer for XNNPACK delegate.
   private var cXNNPackDelegate: Delegate.CDelegate?
 
@@ -121,6 +124,7 @@ public final class Interpreter {
 
     self.options = options
     self.delegates = delegates
+    self._model = model
     options.map {
       if let threadCount = $0.threadCount, threadCount > 0 {
         TfLiteInterpreterOptionsSetNumThreads(cInterpreterOptions, Int32(threadCount))