Vengineerの妄想

人生を妄想しています。

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 

のようです。

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