blabla

2019-07-12 17:42:29 -07:00
parent f36a646ffc
commit c1c7062914
4 changed files with 108 additions and 82 deletions
--- a/examples/python/tensorflow/run.py
+++ b/examples/python/tensorflow/run.py
@@ -58,7 +58,7 @@ def blocksparse_matmul_grad(op, dy):
    return (dx, dw)

 def run_shift():
-    B, C, H, W = 16, 16, 4, 4
+    B, C, H, W = 16, 16, 2, 2
    R, S, F = 3, 3, 32
    stride_h, stride_w = 2, 2
    np.random.seed(2)