Fix broken tests

priv/bin/tensorflow-port

@@ -573,31 +573,31 @@ def test_model_stats():
     assert_equals(s1, "predictions_per_second", None)
     assert_equals(s1, "last_memory_bytes", None)
 
-    s2 = generate_stats([(1, 10000, 100)])
+    s2 = generate_stats([(1, 10000, 100 * 1024)])
     assert_equals(s2, "last_batch_time_ms", 10.0)
     assert_equals(s2, "average_batch_time_ms", 10.0)
     assert_equals(s2, "predictions_per_second", 100.0)
     assert_equals(s2, "last_memory_bytes", 102400)
 
-    s3 = generate_stats([(1, 10000, 100), (1, 10000, 200)])
+    s3 = generate_stats([(1, 10000, 100 * 1024), (1, 10000, 200 * 1024)])
     assert_equals(s3, "last_batch_time_ms", 10.0)
     assert_equals(s3, "average_batch_time_ms", 10.0)
     assert_equals(s3, "predictions_per_second", 100.0)
     assert_equals(s3, "last_memory_bytes", 204800)
 
-    s4 = generate_stats([(1, 10000, 200), (1, 20000, 100)])
+    s4 = generate_stats([(1, 10000, 200 * 1024), (1, 20000, 100 * 1024)])
     assert_equals(s4, "last_batch_time_ms", 20.0)
     assert_equals(s4, "average_batch_time_ms", 50000 / 3 / 1000)
     assert_equals(s4, "predictions_per_second", 1 / (50000 / 3 / 1000000))
     assert_equals(s4, "last_memory_bytes", 102400)
 
     s5 = generate_stats(
-        [(1, 60000, 101),
-         (1, 50000, 102),
-         (1, 40000, 103),
-         (1, 30000, 104),
-         (1, 20000, 105),
-         (1, 10000, 106)])
+        [(1, 60000, 101 * 1024),
+         (1, 50000, 102 * 1024),
+         (1, 40000, 103 * 1024),
+         (1, 30000, 104 * 1024),
+         (1, 20000, 105 * 1024),
+         (1, 10000, 106 * 1024)])
     assert_equals(s5, "last_batch_time_ms", 10.0)
     assert_equals(s5, "average_batch_time_ms", 350000 / 15 / 1000)
     assert_equals(s5, "predictions_per_second", 1 / (350000 / 15 / 1000000))