| 335 | } |
| 336 | |
| 337 | Ref<Op> Graph::addUpsample(const std::string& name, |
| 338 | const Ref<Op>& srcOp) |
| 339 | { |
| 340 | auto srcAlloc = tensorAllocs[srcOp.get()]; |
| 341 | auto op = engine->newUpsample({srcAlloc->desc}); |
| 342 | op->setName(name); |
| 343 | auto dstAlloc = addOp(op, {srcOp}, op->getDstDesc()); |
| 344 | |
| 345 | lazyInits.push_back([=]() |
| 346 | { |
| 347 | op->setSrc(srcAlloc->tensor); |
| 348 | op->setDst(dstAlloc->tensor); |
| 349 | }); |
| 350 | |
| 351 | return op; |
| 352 | } |
| 353 | |
| 354 | void Graph::addOp(const Ref<Op>& op, |
| 355 | const std::vector<Ref<Op>>& srcOps, |
nothing calls this directly
no test coverage detected