add ATen tensor support to pytorch tuple_parser (pytorch#2102)

Author: killeent

torch/csrc/utils/tuple_parser.cpp

@@ -52,6 +52,11 @@ auto TupleParser::parse(std::shared_ptr<thpp::Tensor>& x, const std::string& par
  x.reset(torch::createTensor(obj)->clone_shallow());
 }
 
+auto TupleParser::parse(at::Tensor& x, const std::string& param_name) -> void {
+ PyObject* obj = next_arg();
+ x = torch::createTensorAT(obj);
+}
+
 auto TupleParser::parse(std::vector<int>& x, const std::string& param_name) -> void {
  PyObject* obj = next_arg();
  if (!PyTuple_Check(obj)) {

torch/csrc/utils/tuple_parser.h

@@ -4,6 +4,7 @@
 #include <memory>
 #include <vector>
 #include <THPP/THPP.h>
+#include <ATen/ATen.h>
 
 namespace torch {
 
@@ -15,6 +16,7 @@ struct TupleParser {
  void parse(double& x, const std::string& param_name);
  void parse(std::unique_ptr<thpp::Tensor>& x, const std::string& param_name);
  void parse(std::shared_ptr<thpp::Tensor>& x, const std::string& param_name);
+ void parse(at::Tensor& x, const std::string& param_name);
  void parse(std::vector<int>& x, const std::string& param_name);
  void parse(std::string& x, const std::string& param_name);