IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Convolution: Change element type in onnx unit tests with dyn shapes and convolution nodes

Browse Source
This commit is contained in:
ggalieroc
2021-03-22 11:34:51 +01:00
parent 231ddc6f1b
commit c9f5944b6b
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
ngraph/test/models/onnx/dynamic_shapes/conv_with_dynamic_bias.prototxt
View File

@@ -14,7 +14,7 @@ graph {
name: "data"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_param: "batch"
@@ -36,7 +36,7 @@ graph {
name: "filters"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_param: "feature maps"
@@ -58,7 +58,7 @@ graph {
name: "bias"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_param: "bias"
@@ -71,7 +71,7 @@ graph {
name: "dyn_conv_out"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_param: "batch"

8
ngraph/test/onnx/onnx_import_dyn_shapes.in.cpp
View File

@@ -302,12 +302,12 @@ NGRAPH_TEST(${BACKEND_NAME}, onnx_dyn_shapes_model_conv_with_dynamic_bias)
const auto data_elems = shape_size(data_shape);
const auto filters_elems = shape_size(filters_shape);
test_case.add_input(data_shape, std::vector<int64_t>(data_elems, 1));
test_case.add_input(filters_shape, std::vector<int64_t>(filters_elems, 1));
test_case.add_input(Shape{10}, std::vector<int64_t>(10, 1));
test_case.add_input(data_shape, std::vector<float>(data_elems, 1));
test_case.add_input(filters_shape, std::vector<float>(filters_elems, 1));
test_case.add_input(Shape{10}, std::vector<float>(10, 1));
const auto expected_out_shape = Shape{1, 10, 6, 6};
const std::vector<int64_t> expected_values(shape_size(expected_out_shape), 13);
const std::vector<float> expected_values(shape_size(expected_out_shape), 13);
test_case.add_expected_output(expected_out_shape, expected_values);
test_case.run();