TFRTのソースコードを覗いてみた (その2)

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

TFRTでサポートしてるOpを見ていきます。

対応しているデバイスは、CPUのみで、TensorFlowの中で以下のOpのみサポートしているようです。4つのOpのみです。

• tf.Const op
• tf.AddV2 op
• tf.Matmul op
• tf.Relu op

tf.Const を除いた各Opに対応するKernelが cpu_kernel.h にあります。

• Add
• MatMul2DKernel
• Relu

この他に、image 関連として、

• image.decode_jpeg
• image.resize_bilinear

また、proto なるものもある。

• proto.parse_example_from_bytes
• proto.get_bytes_field_from_example
• proto.get_int64_field_from_example

今後は、TensorFlow v1.x/v2.xでサポートしてる Op をどんどんサポートしていくんでしょうね。