[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>
Wed, 29 Dec 2021 16:52:00 +0000 (11:52 -0500)
commit72f450af5b21bcd94676ddb0f619143cc88b78cc
treecce81a04dd3e00e298119fcf9973680016c925b6
parent9b05ea9a700077531df11395c5147bba3b18a452
[batchnorm] accepts any input size, allocate weights in get_output_shape
src/ai/batchnorm.c
src/ai/batchnorm.h