TensorFlow.NET/src/TensorflowNET.Hub/registry.cs at master · feelsyt/TensorFlow.NET · GitHub

History

55 lines (48 loc) · 1.56 KB

Raw

using System;

using System.Collections.Generic;

using System.Linq;

namespace Tensorflow.Hub

{

internal class MultiImplRegister

{

private static MultiImplRegister resolver = new MultiImplRegister("resolver", new IResolver[0]);

private static MultiImplRegister loader = new MultiImplRegister("loader", new IResolver[0]);

static MultiImplRegister()

{

resolver.add_implementation(new PathResolver());

resolver.add_implementation(new HttpUncompressedFileResolver());

resolver.add_implementation(new GcsCompressedFileResolver());

resolver.add_implementation(new HttpCompressedFileResolver());

}

string _name;

List<IResolver> _impls;

public MultiImplRegister(string name, IEnumerable<IResolver> impls)

{

_name = name;

_impls = impls.ToList();

}

public void add_implementation(IResolver resolver)

{

_impls.Add(resolver);

}

public string Call(string handle)

{

foreach (var impl in _impls.Reverse<IResolver>())

{

if (impl.IsSupported(handle))

{

return impl.Call(handle);

}

throw new RuntimeError($"Cannot resolve the handle {handle}");

}

public static MultiImplRegister GetResolverRegister()

{

return resolver;

}

public static MultiImplRegister GetLoaderRegister()

{

return loader;

}

TensorFlow.NET/src/TensorflowNET.Hub/registry.cs at master · feelsyt/TensorFlow.NET · GitHub

Expand file tree

Breadcrumbs

registry.cs

Latest commit

History

Breadcrumbs

registry.cs

File metadata and controls