BatchNorm -> BatchNormalization

pfnet-research · Oct 16, 2018 · 3472cc2 · 3472cc2
1 parent d395001
commit 3472cc2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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"),
-Original file line number
+Diff line change
@@ Expand Up / @@ -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") {
     {
@@ Expand Down @@