ext/cumo/narray/gen/spec.rb in cumo-0.3.3 vs ext/cumo/narray/gen/spec.rb in cumo-0.3.4
- old
+ new
@@ -58,10 +58,11 @@
def_id "conv"
def_id "conv_transpose"
def_id "conv_grad_w"
def_id "batch_norm"
def_id "batch_norm_backward"
+ def_id "fixed_batch_norm"
def_id "pooling_forward"
def_id "pooling_backward"
end
if is_int && !is_object
@@ -346,9 +347,10 @@
def_method "conv"
def_method "conv_transpose" # conv_backward_data
def_method "conv_grad_w" # conv_backward_filter
def_method "batch_norm"
def_method "batch_norm_backward"
+ def_method "fixed_batch_norm"
def_method "pooling_forward" # max_pool, avg_pool
def_method "pooling_backward"
end
# rmsdev