[dnn] Adding batchnorm

2019-07-08 18:44:37 -07:00
parent b0cf3143c5
commit f9db0449b7
42 changed files with 682 additions and 1763 deletions
--- a/lib/codegen/shmem_allocation.cpp
+++ b/lib/codegen/shmem_allocation.cpp
@@ -29,6 +29,8 @@ unsigned shmem_allocation::is_ld_padded(ir::value *x) {
 }

 unsigned shmem_allocation::get_num_bytes(ir::value *x) {
+  if(dynamic_cast<ir::reduce_inst*>(x))
+    return 32;
  unsigned result = x->get_type()->get_primitive_size_in_bits() / 8;
  unsigned pad = is_ld_padded(x);
  if(pad > 0){