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

Fixed typo in FullyConnected.

Browse Source 
PiperOrigin-RevId: 304690191
Change-Id: I5bde450f0cba6ca64d09e015ba2439e7aa20e5a3
This commit is contained in:
Raman Sarokin 2020-04-03 14:09:47 -07:00 committed by TensorFlower Gardener
parent 512e26cfb1
commit 9a85d6fe42
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/delegates/gpu/metal/kernels/fully_connected.cc
View File

@ -110,7 +110,7 @@ std::string GetFullyConnectedCode(const DeviceInfo& device_info,
const int linear_index = ugid.x / 4; 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]) + 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]; biases[linear_index];
uint3 gid = uint3(1u, 1u, uint(linear_index)); uint3 gid = uint3(0u, 0u, uint(linear_index));
$$2 $$2
result[linear_index] = value; result[linear_index] = value;
} }