spec.rb in cumo-0.3.0.pre1

- old
+ new

@@ -51,10 +51,20 @@
   def_id "<=>","ufo"
 end
 if (is_float || is_complex) && !is_object
   def_id "gemm"
 end
+# cudnn
+if is_float && !is_complex && !is_object
+  def_id "conv"
+  def_id "conv_transpose"
+  def_id "conv_grad_w"
+  def_id "batch_norm"
+  def_id "batch_norm_backward"
+  def_id "pooling_forward"
+  def_id "pooling_backward"
+end
 
 if is_int && !is_object
   def_id "minlength" # for bincount
 end
 
@@ -327,9 +337,20 @@
 
 # dot
 accum_binary "mulsum"
 if (is_float || is_complex) && !is_object
   def_method "gemm"
+end
+
+# cudnn
+if is_float && !is_complex && !is_object
+  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 "pooling_forward" # max_pool, avg_pool
+  def_method "pooling_backward"
 end
 
 # rmsdev
 # prod
 