| 580 | std::shared_ptr<mgb::Hashable> key = nullptr; |
| 581 | |
| 582 | std::shared_ptr<OpDef> build() { |
| 583 | if (key == nullptr) { |
| 584 | key = std::make_shared<UniqueKey>(); |
| 585 | } |
| 586 | return SubgraphOp::make( |
| 587 | name, std::make_shared<Subgraph>(graph), output_grad_mask, key); |
| 588 | } |
| 589 | }; |
| 590 | |
| 591 | py::class_<PySubgraphBuilder>(m, "SubgraphBuilder") |
no test coverage detected