aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/IBackendInternal.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/IBackendInternal.cpp')
-rw-r--r--src/backends/backendsCommon/IBackendInternal.cpp18
1 files changed, 18 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/IBackendInternal.cpp b/src/backends/backendsCommon/IBackendInternal.cpp
index b08dff84ed..31706eb1e7 100644
--- a/src/backends/backendsCommon/IBackendInternal.cpp
+++ b/src/backends/backendsCommon/IBackendInternal.cpp
@@ -76,6 +76,17 @@ IBackendInternal::IWorkloadFactoryPtr IBackendInternal::CreateWorkloadFactory(
return CreateWorkloadFactory(tensorHandleFactoryRegistry);
}
+IBackendInternal::IWorkloadFactoryPtr IBackendInternal::CreateWorkloadFactory(
+ class TensorHandleFactoryRegistry& tensorHandleFactoryRegistry,
+ const ModelOptions& modelOptions,
+ MemorySourceFlags inputFlags,
+ MemorySourceFlags outputFlags) const
+{
+ IgnoreUnused(inputFlags);
+ IgnoreUnused(outputFlags);
+ return CreateWorkloadFactory(tensorHandleFactoryRegistry, modelOptions);
+}
+
IBackendInternal::IBackendContextPtr IBackendInternal::CreateBackendContext(const IRuntime::CreationOptions&) const
{
return IBackendContextPtr{};
@@ -147,6 +158,13 @@ bool IBackendInternal::SupportsTensorAllocatorAPI() const
return !GetHandleFactoryPreferences().empty();
}
+void IBackendInternal::RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry,
+ MemorySourceFlags /*inputFlags*/,
+ MemorySourceFlags /*outputFlags*/)
+{
+ return RegisterTensorHandleFactories(registry);
+}
+
ITensorHandleFactory::FactoryId IBackendInternal::GetBackwardCompatibleFavoriteHandleFactory()
{
auto favorites = GetHandleFactoryPreferences();