| 19 | |
| 20 | protected: |
| 21 | auto graph_eager() { |
| 22 | if (!m_graph_eager) { |
| 23 | m_graph_eager = ComputingGraph::make(); |
| 24 | m_graph_eager->options().eager_evaluation = true; |
| 25 | } |
| 26 | return m_graph_eager.get(); |
| 27 | } |
| 28 | void make_graph_normal( |
| 29 | const SmallVector<std::shared_ptr<HostTensorND>>& inputs, |
| 30 | const thin_function<SymbolVarArray(const SymbolVarArray&)>& make_graph, |