| 21 | fastMath(fastMath) {} |
| 22 | |
| 23 | Ref<InputProcess> Graph::addInputProcess(const std::string& name, |
| 24 | const TensorDims& srcDims, |
| 25 | const std::shared_ptr<TransferFunction>& transferFunc, |
| 26 | bool hdr, |
| 27 | bool snorm) |
| 28 | { |
| 29 | auto op = engine->newInputProcess({srcDims, transferFunc, hdr, snorm}); |
| 30 | op->setName(name); |
| 31 | auto dstAlloc = addOp(op, {}, op->getDstDesc()); |
| 32 | |
| 33 | lazyInits.push_back([=]() |
| 34 | { |
| 35 | op->setDst(dstAlloc->tensor); |
| 36 | }); |
| 37 | |
| 38 | return op; |
| 39 | } |
| 40 | |
| 41 | Ref<OutputProcess> Graph::addOutputProcess(const std::string& name, |
| 42 | const Ref<Op>& srcOp, |
no test coverage detected