提交 195599ae 编写于 作者: D dolphin8

fix batchnorm

上级 b0c3e1cb
......@@ -70,7 +70,7 @@ kernel void batchnorm(texture2d_array<half, access::read> 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<half, access::read> inTexture [[texture(0)]],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册