【发布时间】:2017-03-09 04:05:55
【问题描述】:
NNpack 是否在 tensorflow 中实现?我发现它可以被更高级别的框架所利用,例如 Torch、Caffe、Tensorflow、Theano 和 Mocha.jl,但是我在 tensorflow 中没有找到任何实现。
【问题讨论】:
标签: tensorflow
NNpack 是否在 tensorflow 中实现?我发现它可以被更高级别的框架所利用,例如 Torch、Caffe、Tensorflow、Theano 和 Mocha.jl,但是我在 tensorflow 中没有找到任何实现。
【问题讨论】:
标签: tensorflow
TensorFlow 目前不使用NNPACK 来加速其数学内核;相反,它使用Eigen::Tensor、cuDNN、cuBLAS、XLA-generated 和手写内核的组合。
如果您想在 TensorFlow 操作中使用 NNPACK,您可以通过使用该库的 adding a custom op kernel 来执行此操作。
【讨论】: