Commit d7c4dedf authored by Jayaram Bobba's avatar Jayaram Bobba

Style fix

parent d66b386d
...@@ -5619,20 +5619,28 @@ TEST(${BACKEND_NAME}, zero_sized_subtract) ...@@ -5619,20 +5619,28 @@ TEST(${BACKEND_NAME}, zero_sized_subtract)
make_binary_empty_test<op::Subtract>("${BACKEND_NAME}"); make_binary_empty_test<op::Subtract>("${BACKEND_NAME}");
} }
TEST (${BACKEND_NAME}, convolution_outlining) TEST(${BACKEND_NAME}, convolution_outlining)
{ {
auto shape_a = Shape{1,2,2,2}; auto shape_a = Shape{1, 2, 2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape_a); auto A = make_shared<op::Parameter>(element::f32, shape_a);
auto shape_b = Shape{2,2,1,1}; auto shape_b = Shape{2, 2, 1, 1};
auto B = make_shared<op::Parameter>(element::f32, shape_b); auto B = make_shared<op::Parameter>(element::f32, shape_b);
auto shape_r = Shape{1,2,2,2}; auto shape_r = Shape{1, 2, 2, 2};
auto conv1 = make_shared<op::Convolution>(A, B, Strides{1,1}, Strides{1, 1}, auto conv1 = make_shared<op::Convolution>(A,
CoordinateDiff{0,0}, CoordinateDiff{0,0}, B,
Strides{1,1}); Strides{1, 1},
auto conv2 = make_shared<op::Convolution>(conv1, B, Strides{1,1}, Strides{1, 1}, Strides{1, 1},
CoordinateDiff{0,0}, CoordinateDiff{0,0}, CoordinateDiff{0, 0},
Strides{1,1}); CoordinateDiff{0, 0},
auto f = make_shared<Function>(conv2, op::Parameters{A,B}); Strides{1, 1});
auto conv2 = make_shared<op::Convolution>(conv1,
B,
Strides{1, 1},
Strides{1, 1},
CoordinateDiff{0, 0},
CoordinateDiff{0, 0},
Strides{1, 1});
auto f = make_shared<Function>(conv2, op::Parameters{A, B});
auto manager = runtime::Manager::get("${BACKEND_NAME}"); auto manager = runtime::Manager::get("${BACKEND_NAME}");
auto external = manager->compile(f); auto external = manager->compile(f);
...@@ -5641,12 +5649,12 @@ TEST (${BACKEND_NAME}, convolution_outlining) ...@@ -5641,12 +5649,12 @@ TEST (${BACKEND_NAME}, convolution_outlining)
// Create some tensors for input/output // Create some tensors for input/output
auto a = backend->make_primary_tensor_view(element::f32, shape_a); auto a = backend->make_primary_tensor_view(element::f32, shape_a);
copy_data(a, vector<float>{1.0f,1.0f,1.0f,1.0f,1.0f,1.0f,1.0f,1.0f}); copy_data(a, vector<float>{1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f});
auto b = backend->make_primary_tensor_view(element::f32, shape_b); auto b = backend->make_primary_tensor_view(element::f32, shape_b);
copy_data(b, vector<float>{1.0f,1.0f,1.0f,1.0f}); copy_data(b, vector<float>{1.0f, 1.0f, 1.0f, 1.0f});
auto result = backend->make_primary_tensor_view(element::f32, shape_r); auto result = backend->make_primary_tensor_view(element::f32, shape_r);
vector<float> expected_result{4.0f,4.0f,4.0f,4.0f,4.0f,4.0f,4.0f,4.0f}; vector<float> expected_result{4.0f, 4.0f, 4.0f, 4.0f, 4.0f, 4.0f, 4.0f, 4.0f};
cf->call({a, b}, {result}); cf->call({a, b}, {result});
EXPECT_EQ(vector<float>{expected_result}, read_vector<float>(result)); EXPECT_EQ(vector<float>{expected_result}, read_vector<float>(result));
......
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