Formatting

48187e79 · turneram · 6202ea15 · 48187e79
Commit 48187e79 authored May 20, 2022 by turneram
Show whitespace changes
Inline Side-by-side

Showing with 9 additions and 5 deletions

test/onnx/gen_onnx.py test/onnx/gen_onnx.py +9 -5

No files found.
--- a/test/onnx/gen_onnx.py
+++ b/test/onnx/gen_onnx.py
@@ -190,11 +190,15 @@ def atanh_test():

 @onnx_test
 def attention_test():
-    input = helper.make_tensor_value_info('input', TensorProto.FLOAT, [2, 384, 768])
-    weights = helper.make_tensor_value_info('weights', TensorProto.FLOAT, [768, 2304])
+    input = helper.make_tensor_value_info('input', TensorProto.FLOAT,
+                                          [2, 384, 768])
+    weights = helper.make_tensor_value_info('weights', TensorProto.FLOAT,
+                                            [768, 2304])
    bias = helper.make_tensor_value_info('bias', TensorProto.FLOAT, [2304])
-    mask_index = helper.make_tensor_value_info('mask_index', TensorProto.INT64, [2, 384])
-    result = helper.make_tensor_value_info('result', TensorProto.FLOAT, [2, 384, 768])
+    mask_index = helper.make_tensor_value_info('mask_index', TensorProto.INT64,
+                                               [2, 384])
+    result = helper.make_tensor_value_info('result', TensorProto.FLOAT,
+                                           [2, 384, 768])

    node = helper.make_node('Attention',
                            inputs=['input', 'weights', 'bias', 'mask_index'],