2017-01-24 6 views

答えて

0

TensorFlowは現在、数学カーネルを高速化するためにNNPACKを使用していません。代わりに、Eigen::Tensor,cuDNN,cuBLAS,XLA-generatedと手書きのカーネルの組み合わせを使用します。

TensorFlowオペレーションでNNPACKを使用する場合は、そのライブラリを使用したadding a custom op kernelでこれを行うことができます。

関連する問題