| 33 | } |
| 34 | |
| 35 | void create_graph() { |
| 36 | input_tensor = m_gen({1, 3, 32, 32}, m_cn); |
| 37 | auto input = opr::Host2DeviceCopy::make(*m_network->graph, input_tensor, m_cn) |
| 38 | .rename("input"); |
| 39 | auto f = m_network->add_conv( |
| 40 | input, 4, {3, 3}, dtype::Float32(), true, {2, 2}, {0, 0}); |
| 41 | f = m_network->add_elemwise( |
| 42 | {f}, dtype::Float32(), opr::Elemwise::Param::Mode::EXP); |
| 43 | f = m_network->add_conv(f, 8, {3, 3}, dtype::Float32(), true, {1, 1}, {1, 1}); |
| 44 | f = m_network->add_pooling(f, {2, 2}, {2, 2}); |
| 45 | m_out_var = m_network->add_concat(f, -f); |
| 46 | } |
| 47 | |
| 48 | void create_relayout_out_graph(int mem_forward_opr_type) { |
| 49 | input_tensor = m_gen({1, 3, 32, 32}, m_cn); |
no test coverage detected