Skip to content

GitLab

Commit b2c6106c authored by Paul's avatar Paul
Browse files

Fix formatting

parent c701d117
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 6 deletions
test/onnx/gen_onnx.py
View file @ b2c6106c
...@@ -6413,19 +6413,20 @@ def slice_test(): ...@@ -6413,19 +6413,20 @@ def slice_test():
return ([node], [x], [y]) return ([node], [x], [y])
@onnx_test() @onnx_test()
def slice_constant_test(): def slice_constant_test():
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 2])
x_tensor = helper.make_tensor(name='x_tensor', x_tensor = helper.make_tensor(name='x_tensor',
data_type=TensorProto.FLOAT, data_type=TensorProto.FLOAT,
dims=[3, 2], dims=[3, 2],
vals=[0, 1, 2, 3, 4, 5]) vals=[0, 1, 2, 3, 4, 5])
x = onnx.helper.make_node('Constant', x = onnx.helper.make_node('Constant',
inputs=[], inputs=[],
outputs=['x'], outputs=['x'],
value=x_tensor) value=x_tensor)
node = onnx.helper.make_node('Slice', node = onnx.helper.make_node('Slice',
inputs=['x'], inputs=['x'],
...@@ -6436,6 +6437,7 @@ def slice_constant_test(): ...@@ -6436,6 +6437,7 @@ def slice_constant_test():
return ([x, node], [], [y]) return ([x, node], [], [y])
@onnx_test() @onnx_test()
def slice_dyn_test(): def slice_dyn_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, None, 2]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, None, 2])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment