Prepare templates

Author: izdeby

tools/autograd/templates/python_torch_functions_dispatch.h

@@ -22,6 +22,7 @@ using at::TensorList;
 using at::IntArrayRef;
 using at::Generator;
 using at::Storage;
+using at::Layout;
 
 ${py_method_dispatch}
 

tools/autograd/templates/python_variable_methods_dispatch.h

@@ -19,6 +19,7 @@ using at::TensorList;
 using at::IntArrayRef;
 using at::Generator;
 using at::Storage;
+using at::Layout;
 
 ${py_method_dispatch}
 

tools/autograd/templates/variable_factories.h

@@ -16,6 +16,10 @@
 #include <initializer_list>
 #include <utility>
 
+using at::Device;
+using at::ScalarType;
+using at::Layout;
+
 #ifdef BUILD_NAMEDTENSOR
 using at::DimnameList;
 #endif