Unverified Commit 08650844 authored by Nick Korovaiko's avatar Nick Korovaiko Committed by GitHub

Update reshape_sinking.cpp

parent 8de62221
...@@ -192,20 +192,6 @@ TEST(reshape_sinking, slice_pad) ...@@ -192,20 +192,6 @@ TEST(reshape_sinking, slice_pad)
auto reshape4 = make_shared<op::Reshape>(avgpool, to_nhwc, Shape{1, 1, 2, 2}); auto reshape4 = make_shared<op::Reshape>(avgpool, to_nhwc, Shape{1, 1, 2, 2});
auto f = make_shared<Function>(reshape4, ParameterVector{A}); auto f = make_shared<Function>(reshape4, ParameterVector{A});
// auto reshape1 = make_shared<op::Reshape>(A, to_nchw, Shape{1, 1, 4, 4});
// auto pad = make_shared<op::Pad>(reshape1, pad_value, padding_below, padding_above, padding_interior);
// auto reshape2 = make_shared<op::Reshape>(pad, to_nhwc, Shape{1, 5, 5, 1});
// auto absn = make_shared<op::Abs>(reshape2);
// auto reshape3 = make_shared<op::Reshape>(absn, to_nchw, Shape{1, 1, 5, 5});
// auto slice = make_shared<op::Slice>(
// reshape3, Coordinate{0, 0, 1, 1}, Coordinate{1, 1, 5, 5}, Strides{1, 1, 1, 1});
// auto reshape4 = make_shared<op::Reshape>(slice, to_nhwc, Shape{1, 4, 4, 1});
// auto absn2 = make_shared<op::Abs>(reshape4);
// auto reshape5 = make_shared<op::Reshape>(absn2, to_nchw, Shape{1, 1, 4, 4});
// auto avgpool = make_shared<op::AvgPool>(reshape5, Shape{1, 1}, Strides{2, 2});
// auto reshape6 = make_shared<op::Reshape>(avgpool, to_nhwc, Shape{1, 1, 2, 2});
// auto f = make_shared<Function>(reshape6, ParameterVector{A});
pass::Manager pass_manager; pass::Manager pass_manager;
size_t before_count = count_ops_of_type<op::Reshape>(f); size_t before_count = count_ops_of_type<op::Reshape>(f);
pass_manager.register_pass<pass::VisualizeTree>("before.pdf"); pass_manager.register_pass<pass::VisualizeTree>("before.pdf");
......
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