commit 381c6e98d2b0429abe66ee53a48a03dd5d868447
author A. Unique TensorFlower <gardener@tensorflow.org> Wed Jul 15 16:31:29 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Wed Jul 15 16:41:28 2020 -0700
tree f15f161b08196b7d9e09e740cd47b3738d945dc7
parent c2f5a7f68da6a08db1610d79bd427ab564918ba8

Use internal linkage for classes defined in compute_task.mm.

PiperOrigin-RevId: 321461539
Change-Id: Ie6b9755897088b5363c62d9ab8e8d93ca89af9a8

tensorflow/lite/delegates/gpu/metal/compute_task.mm

diff --git a/tensorflow/lite/delegates/gpu/metal/compute_task.mm b/tensorflow/lite/delegates/gpu/metal/compute_task.mm
index 88be867..7bfbb55 100644
--- a/tensorflow/lite/delegates/gpu/metal/compute_task.mm
+++ b/tensorflow/lite/delegates/gpu/metal/compute_task.mm
@@ -39,22 +39,28 @@
 using ::tflite::gpu::uint3;
 using ::tflite::gpu::ValueId;
 
-@implementation TFLComputeTask {
-  struct InputBuffer {
-    ValueId uid;
-    id<MTLBuffer> metalHandle;
-  };
-  struct OutputBuffer {
-    ValueId uid;
-    id<MTLBuffer> metalHandle;
-    OutputDimensions dimensionsFunction;
-    std::vector<ValueId> alias;
-  };
-  struct UniformBuffer {
-    std::vector<uint8_t> data;
-    UniformsFunction dataFunction;
-  };
+namespace {
 
+struct InputBuffer {
+  ValueId uid;
+  id<MTLBuffer> metalHandle;
+};
+
+struct OutputBuffer {
+  ValueId uid;
+  id<MTLBuffer> metalHandle;
+  OutputDimensions dimensionsFunction;
+  std::vector<ValueId> alias;
+};
+
+struct UniformBuffer {
+  std::vector<uint8_t> data;
+  UniformsFunction dataFunction;
+};
+
+}  // namespace
+
+@implementation TFLComputeTask {
   id<MTLComputePipelineState> _program;
   std::vector<InputBuffer> _inputBuffers;
   std::vector<OutputBuffer> _outputBuffers;