| 38 | |
| 39 | MIGRAPHX_DECLARE_ENV_VAR(MIGRAPHX_REWRITE_LRN); |
| 40 | static void opt_pooling(migraphx::module& m) |
| 41 | { |
| 42 | migraphx::rewrite_pooling rp; |
| 43 | migraphx::dead_code_elimination dce; |
| 44 | rp.apply(m); |
| 45 | dce.apply(m); |
| 46 | } |
| 47 | |
| 48 | TEST_CASE(rewrite_pooling_test) |
| 49 | { |