人気ブログランキング | 話題のタグを見る

雑多な趣味の記録帳

tom01h.exblog.jp

tiny-dnn アクセラレータのシーケンサ追加

tiny-dnn アクセラレータに 1サンプル分のシーケンサを追加しました。
カーネルごとに入力データのアドレスを生成してデータを取ってきつつ計算し、出力データのアドレスを生成してデータを吐き出します。
終わったらストライド分(1限定だけど)だけ移動して、1サンプルの計算が終わるまで繰り返します。
ここまでは自動で動くようになりました。

次は 1サンプル分のデータ用バッファを内蔵して、AXI-Stream っぽいインタフェースを追加したいと思います。

しかし、ごちゃごちゃしたシーケンサを作ってしまったため、逆伝搬への対応がすごく難しい事になりそうな予感です…

by tom01h | 2018-10-22 23:57