Fixed typo in FullyConnected.

PiperOrigin-RevId: 304690191 Change-Id: I5bde450f0cba6ca64d09e015ba2439e7aa20e5a3
2020-04-03 14:09:47 -07:00 · 2020-04-03 14:09:47 -07:00 · 9a85d6fe42
commit 9a85d6fe42
parent 512e26cfb1
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/lite/delegates/gpu/metal/kernels/fully_connected.cc
+++ b/tensorflow/lite/delegates/gpu/metal/kernels/fully_connected.cc
@ -110,7 +110,7 @@ std::string GetFullyConnectedCode(const DeviceInfo& device_info,
    const int linear_index = ugid.x / 4;
    FLT4 value = FLT4(temp[tid.x][0], temp[tid.x + 1][0], temp[tid.x + 2][0], temp[tid.x + 3][0]) +
      biases[linear_index];
-    uint3 gid = uint3(1u, 1u, uint(linear_index));
+    uint3 gid = uint3(0u, 0u, uint(linear_index));
    $$2
    result[linear_index] = value;
  }