Skip to content

GitLab

Commit c6a7d7cb authored by Brian Pickrell's avatar Brian Pickrell
Browse files

formatting

parent 6ad1e633
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
test/onnx/gen_onnx.py
View file @ c6a7d7cb
...@@ -4433,7 +4433,8 @@ def multinomial_test(): ...@@ -4433,7 +4433,8 @@ def multinomial_test():
def multinomial_dyn_test(): def multinomial_dyn_test():
sample_size = 13 sample_size = 13
seed = 0.0 seed = 0.0
input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [None, 10]) input = helper.make_tensor_value_info("input", TensorProto.FLOAT,
[None, 10])
output = helper.make_tensor_value_info("output", TensorProto.INT32, output = helper.make_tensor_value_info("output", TensorProto.INT32,
[None, 10]) [None, 10])
...@@ -4450,7 +4451,8 @@ def multinomial_dyn_test(): ...@@ -4450,7 +4451,8 @@ def multinomial_dyn_test():
def multinomial_autoseed_dyn_test(): def multinomial_autoseed_dyn_test():
# If seed attribute is not given, device should auto generate one at runtime # If seed attribute is not given, device should auto generate one at runtime
sample_size = 12 sample_size = 12
input = helper.make_tensor_value_info("input", TensorProto.FLOAT, [None, 10]) input = helper.make_tensor_value_info("input", TensorProto.FLOAT,
[None, 10])
output = helper.make_tensor_value_info("output", TensorProto.INT32, output = helper.make_tensor_value_info("output", TensorProto.INT32,
[None, 10]) [None, 10])
......
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