diff --git a/metal/paddle-mobile/paddle-mobile/Operators/Kernels/Kernels.metal b/metal/paddle-mobile/paddle-mobile/Operators/Kernels/Kernels.metal index 82b9fa2ab04e4107dac64997a77a4eae5c4cdcec..54337e7c7c2477a0595ad36a6089d247fd9503f9 100644 --- a/metal/paddle-mobile/paddle-mobile/Operators/Kernels/Kernels.metal +++ b/metal/paddle-mobile/paddle-mobile/Operators/Kernels/Kernels.metal @@ -70,7 +70,7 @@ kernel void batchnorm(texture2d_array inTexture [[texture(0) gid.z >= outTexture.get_array_size()) return; const half4 input = inTexture.read(gid.xy, gid.z); half4 output = input * newScale[gid.z] + newBias[gid.z]; - outTexture.write(input, gid.xy, gid.z); + outTexture.write(output, gid.xy, gid.z); } kernel void texture2d_to_2d_array(texture2d inTexture [[texture(0)]],