| 26 | } |
| 27 | |
| 28 | std::shared_ptr<TensorMap> Subdevice::getCachedTensors(const void* key) |
| 29 | { |
| 30 | std::shared_ptr<TensorMap>& tensorMap = cachedTensors[key]; |
| 31 | if (!tensorMap) |
| 32 | tensorMap = std::make_shared<TensorMap>(); |
| 33 | return tensorMap; |
| 34 | } |
| 35 | |
| 36 | OIDN_NAMESPACE_END |