| 29 | AutoOprChecker<nr_inp, nr_out, dtype>& AutoOprChecker<nr_inp, nr_out, dtype> |
| 30 | |
| 31 | DEF_IMPL()::AutoOprChecker(GraphMaker maker, FwdNumeric fwd, CompNode comp_node) |
| 32 | : m_fwd(fwd), m_maker(maker), m_comp_node{comp_node} { |
| 33 | InputGenerator default_input_gen = [this](HostTensorND& dest) { |
| 34 | dest = *m_gen(dest.shape(), m_comp_node); |
| 35 | }; |
| 36 | for (size_t i = 0; i < nr_inp; ++i) { |
| 37 | m_inputs[i] = std::make_shared<HostTensorND>(m_comp_node, dtype()); |
| 38 | m_inputs_generator[i] = default_input_gen; |
| 39 | } |
| 40 | for (size_t i = 0; i < nr_inp; ++i) { |
| 41 | m_inputs_allow_grad[i] = true; |
| 42 | } |
| 43 | for (size_t i = 0; i < nr_out; ++i) { |
| 44 | m_outputs_allow_grad[i] = true; |
| 45 | } |
| 46 | for (size_t i = 0; i < nr_out; ++i) { |
| 47 | m_outputs_allow_check[i] = true; |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | DEF_IMPL(void)::build_graph() { |
| 52 | mgb_assert(!m_built); |
nothing calls this directly
no test coverage detected