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

Convolution: Change onnx test with dynamic shapes to float element type

Browse Source
This commit is contained in:
ggalieroc 2021-02-17 16:52:03 +01:00
parent 3073968218
commit 1f4202b010
2 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ngraph/test/models/onnx/dynamic_shapes/conv_with_dynamic_batch.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_value: 10
@ -58,7 +58,7 @@ graph {
name: "bias"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_value: 10
@ -71,7 +71,7 @@ graph {
name: "dyn_conv_out"
type {
tensor_type {
elem_type: 7
elem_type: 1
shape {
dim {
dim_param: "batch"

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

@ -279,13 +279,13 @@ NGRAPH_TEST(${BACKEND_NAME}, onnx_dyn_shapes_model_conv_with_dynamic_batch)
const auto data_elems = shape_size(data_shape);
const auto filters_elems = shape_size(filters_shape);
test_case.add_input<int64_t>(data_shape, std::vector<int64_t>(data_elems, 1));
test_case.add_input<int64_t>(filters_shape, std::vector<int64_t>(filters_elems, 1));
test_case.add_input<int64_t>(Shape{10}, std::vector<int64_t>(10, 1));
test_case.add_input<float>(data_shape, std::vector<float>(data_elems, 1));
test_case.add_input<float>(filters_shape, std::vector<float>(filters_elems, 1));
test_case.add_input<float>(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);
test_case.add_expected_output<int64_t>(expected_out_shape, expected_values);
const std::vector<float> expected_values(shape_size(expected_out_shape), 13);
test_case.add_expected_output<float>(expected_out_shape, expected_values);
test_case.run();
}