-

diff --git a/tools/autograd/gen_variable_type.py b/tools/autograd/gen_variable_type.py
index 3378d3b..05ac18f 100644
--- a/tools/autograd/gen_variable_type.py
+++ b/tools/autograd/gen_variable_type.py
@@ -401,7 +401,7 @@ def preprocess_nn_functions(declarations):
     autograd_functions = []
     for declaration in declarations:
         name = declaration['name']
-        if name == 'batch_norm' or 'conv' in name:
+        if name == 'batch_norm' or 'conv3d' in name:
             continue
 
         fwd_name = name + '_forward'