Skip to content

GitLab

Commit 38f0d386 authored by charlie's avatar charlie
Browse files

formatting

parent 2dff4dd2
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 11 deletions
test/onnx/gen_onnx.py
View file @ 38f0d386
...@@ -827,6 +827,7 @@ def conv_bn_relu_maxpool_test(): ...@@ -827,6 +827,7 @@ def conv_bn_relu_maxpool_test():
return ([node0, node1, node2, node3], [x, y, z, m, n, k, l], [out]) return ([node0, node1, node2, node3], [x, y, z, m, n, k, l], [out])
@onnx_test @onnx_test
def conv_dynamic_batch_test(): def conv_dynamic_batch_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [None, 3, 5, 5])
...@@ -840,7 +841,8 @@ def conv_dynamic_batch_test(): ...@@ -840,7 +841,8 @@ def conv_dynamic_batch_test():
@onnx_test @onnx_test
def conv_dynamic_img_test(): def conv_dynamic_img_test():
x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, None, None]) x = helper.make_tensor_value_info('0', TensorProto.FLOAT,
[1, 3, None, None])
y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3]) y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 3, 3, 3])
out = helper.make_tensor_value_info('2', TensorProto.FLOAT, out = helper.make_tensor_value_info('2', TensorProto.FLOAT,
[1, 1, None, None]) [1, 1, None, None])
......
test/onnx/onnx_test.cpp
View file @ 38f0d386
...@@ -777,7 +777,8 @@ TEST_CASE(conv_dynamic_batch_test) ...@@ -777,7 +777,8 @@ TEST_CASE(conv_dynamic_batch_test)
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
auto l0 = mm->add_parameter("0", {migraphx::shape::float_type, {{1, 6, 0}, {3, 3, 0}, {5, 5, 0}, {5, 5, 0}}}); auto l0 = mm->add_parameter(
"0", {migraphx::shape::float_type, {{1, 6, 0}, {3, 3, 0}, {5, 5, 0}, {5, 5, 0}}});
auto l1 = mm->add_parameter("1", {migraphx::shape::float_type, {1, 3, 3, 3}}); auto l1 = mm->add_parameter("1", {migraphx::shape::float_type, {1, 3, 3, 3}});
auto c0 = mm->add_instruction( auto c0 = mm->add_instruction(
migraphx::make_op("convolution", migraphx::make_op("convolution",
...@@ -797,7 +798,8 @@ TEST_CASE(conv_dynamic_img_test) ...@@ -797,7 +798,8 @@ TEST_CASE(conv_dynamic_img_test)
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
auto l0 = mm->add_parameter("0", {migraphx::shape::float_type, {{1, 1, 0}, {3, 3, 0}, {5, 10, 0}, {5, 10, 0}}}); auto l0 = mm->add_parameter(
"0", {migraphx::shape::float_type, {{1, 1, 0}, {3, 3, 0}, {5, 10, 0}, {5, 10, 0}}});
auto l1 = mm->add_parameter("1", {migraphx::shape::float_type, {1, 3, 3, 3}}); auto l1 = mm->add_parameter("1", {migraphx::shape::float_type, {1, 3, 3, 3}});
auto c0 = mm->add_instruction( auto c0 = mm->add_instruction(
migraphx::make_op("convolution", migraphx::make_op("convolution",
......
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