TensorFlow Liteに、なんか、追加されている

@Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった 

TensorFlow Liteのexperimentalに、

「Accelerator whitelisting」

なるものがアップされていた。

github.com

TensorFlow Delagateがいっぱいあるんですが、微妙にAPIが違うんです。

ということで、

• NNAPI
• GPU
• Hexagon

に関しては、同じAPIで引数の文字列で切り替えるようにしたもの。

この3つということは、Snapdragonのスマホ用なのかな。。。。

void CreateDelegate(flatbuffers::Offset<NNAPISettings> settings) {
    settings_ = flatbuffers::GetTemporaryPointer(
        fbb_, CreateTFLiteSettings(fbb_, tflite::Delegate_NNAPI, settings));

    plugin_ = delegates::DelegatePluginRegistry::CreateByName("NnapiPlugin",
*settings_);
    delegate_ = plugin_->Create();
}

 こんな感じで Delegate を生成するのね。名前は、クラス名の模様。

• NNAPI         => NnapiPlugin
• GPU             => GpuPlugin
• Hexagon      => HexagonPlugin 

のようです。

• flex
• nnpack

は、どうなるのだろうか？