Vengineerの戯言

人生は短いけど、長いです。人生を楽しみましょう!

TensorFlow XLA/TPUソースコード解析

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

TensorFlow XLA/TPUのソースコード解析のアンケート。

39票中、約90%ということで、35票は「是非、みたい」ということなので、公開します。

 

ただし、公開は1週間程度になります。たぶん。

こちら、↓。

docs.google.com

 

TensorFlow => XLA => MLIR => TPU Program になっていて、

ソースコードとしては、MLIR まで、MLIR => TPU Program の部分は、

tf._TPUCompileMLIR になっていて、この部分は公開されていません。

それ以外の TPUxxxx は、Dialect 'tf’ definition にも説明がありますね。

www.tensorflow.org