From 605e3a93632f92458987904da0ccef74915df647 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Sun, 30 Oct 2016 00:20:15 +0100
Subject: [PATCH] don't rebatch batches

---
 src/backend/tensorflow/model.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/backend/tensorflow/model.jl b/src/backend/tensorflow/model.jl
index 480eb7f7..eb7d1b69 100644
--- a/src/backend/tensorflow/model.jl
+++ b/src/backend/tensorflow/model.jl
@@ -20,6 +20,7 @@ function tf(model)
 end
 
 batchone(x) = Batch((x,))
+batchone(x::Batch) = x
 
 function batch(xs)
   dims = ndims(xs)-1