[batchnorm] accepts any input size, allocate weights in get_output_shape
authorPaul Brossier <piem@piem.org>
Fri, 18 Jan 2019 09:47:14 +0000 (10:47 +0100)
committerPaul Brossier <piem@piem.org>
Fri, 18 Jan 2019 09:47:14 +0000 (10:47 +0100)
commit689ba93b06ffce012e4a45ed0637e88d44e9c97d
treeb8b758820a1617696a62442be443dbc338588317
parent3977b4f46c2b4ffcbe78d3dce446cf21bad2c207
[batchnorm] accepts any input size, allocate weights in get_output_shape
src/ai/batchnorm.c
src/ai/batchnorm.h