From 3472cc2073c0f6eaac91b4b1482b545af12071c8 Mon Sep 17 00:00:00 2001
From: okdshin <kokuzen@gmail.com>
Date: Tue, 16 Oct 2018 20:39:46 +0900
Subject: [PATCH] BatchNorm -> BatchNormalization

---
 menoh/attribute_completion_and_shape_inference.hpp          | 2 +-
 scripts/gen_attribute_completion_and_shape_inference_hpp.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/menoh/attribute_completion_and_shape_inference.hpp b/menoh/attribute_completion_and_shape_inference.hpp
index 452afd7..29a6e7b 100644
--- a/menoh/attribute_completion_and_shape_inference.hpp
+++ b/menoh/attribute_completion_and_shape_inference.hpp
@@ -175,7 +175,7 @@ add_variable_to_table(output(0), dtype_of(input(0)),
 else
 
 
-if(node.op_type == "BatchNorm") {
+if(node.op_type == "BatchNormalization") {
     
     
 {
diff --git a/scripts/gen_attribute_completion_and_shape_inference_hpp.py b/scripts/gen_attribute_completion_and_shape_inference_hpp.py
index 18861f5..aa5b677 100644
--- a/scripts/gen_attribute_completion_and_shape_inference_hpp.py
+++ b/scripts/gen_attribute_completion_and_shape_inference_hpp.py
@@ -161,7 +161,7 @@ def main():
 assert(2 <= ndims_of(input(0)));
 '''))
     code_list.append(
-        make_completion_code("BatchNorm", [
+        make_completion_code("BatchNormalization", [
             ("epsilon", "float", "1.e-05f"),
             ("momentum", "float", "0.9f"),
             ("spatial", "int", "1"),