commit 52e0816bedf11382e988f313acaa0493938e2dd7
author Trevor Killeen <killeentm@gmail.com> Thu Aug 17 10:28:58 2017 -0700
committer Soumith Chintala <soumith@gmail.com> Tue Sep 05 17:48:55 2017 -0400
tree 544d34591b4c42a0f7579013eea0050b5ff8a42b
parent 72a7530023cc9797f875669037123249ec2e7842

primspec for concat

torch/autograd/_functions/tensor.py

diff --git a/torch/autograd/_functions/tensor.py b/torch/autograd/_functions/tensor.py
index dee17f0..1c214ba 100644
--- a/torch/autograd/_functions/tensor.py
+++ b/torch/autograd/_functions/tensor.py
@@ -320,6 +320,10 @@
 class Concat(Function):
 
     @staticmethod
+    def primspec(dim, *inputs):
+        return torch.toffee.op("Concat", inputs, axis=dim)
+
+    @staticmethod
     def forward(ctx, dim, *inputs):
         ctx.dim = dim
         ctx.input_sizes = [i.size(dim) for i in inputs]