| 37 | } |
| 38 | |
| 39 | ITensorHandleFactory* TensorHandleFactoryRegistry::GetFactory(ITensorHandleFactory::FactoryId id) const |
| 40 | { |
| 41 | for (auto& factory : m_Factories) |
| 42 | { |
| 43 | if (factory->GetId() == id) |
| 44 | { |
| 45 | return factory.get(); |
| 46 | } |
| 47 | } |
| 48 | |
| 49 | return nullptr; |
| 50 | } |
| 51 | |
| 52 | ITensorHandleFactory* TensorHandleFactoryRegistry::GetFactory(ITensorHandleFactory::FactoryId id, |
| 53 | MemorySource memSource) const |