diff options
author | 2018-07-26 09:55:19 -0700 | |
---|---|---|
committer | 2018-07-26 09:55:19 -0700 | |
commit | 8ae93f93c0d7df9ad9b143b5d7c888026759ab85 (patch) | |
tree | 64c25d44ee546c36ff77ee135f27c7e0e940df32 | |
parent | d0d12963008ca3e4a63f6f55432fd353ed0b6be6 (diff) |
Fix formatting
-rw-r--r-- | tensorflow/contrib/tensorrt/test/memory_alignment_test.py | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/tensorflow/contrib/tensorrt/test/memory_alignment_test.py b/tensorflow/contrib/tensorrt/test/memory_alignment_test.py index 227a0ee6ae..3dd95c6f62 100644 --- a/tensorflow/contrib/tensorrt/test/memory_alignment_test.py +++ b/tensorflow/contrib/tensorrt/test/memory_alignment_test.py @@ -39,18 +39,12 @@ class MemoryAlignmentTest(trt_test.TfTrtIntegrationTestBase): g = ops.Graph() with g.as_default(): inp = array_ops.placeholder( - dtype=dtype, - shape=[None] + input_dims[1:], - name=input_name) + dtype=dtype, shape=[None] + input_dims[1:], name=input_name) with g.device("/GPU:0"): e1 = constant_op.constant( - np.random.randn(1, 1, 3, 5), - name="kernel_1", - dtype=dtype) + np.random.randn(1, 1, 3, 5), name="kernel_1", dtype=dtype) e2 = constant_op.constant( - np.random.randn(1, 1, 5, 10), - name="kernel_2", - dtype=dtype) + np.random.randn(1, 1, 5, 10), name="kernel_2", dtype=dtype) conv = nn.conv2d( input=inp, filter=e1, |