【问题标题】:Does NNpack are implemented in tensorflow?NNpack 是否在 tensorflow 中实现?
【发布时间】:2017-03-09 04:05:55
【问题描述】:

NNpack 是否在 tensorflow 中实现?我发现它可以被更高级别的框架所利用,例如 Torch、Caffe、Tensorflow、Theano 和 Mocha.jl,但是我在 tensorflow 中没有找到任何实现。

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    TensorFlow 目前不使用NNPACK 来加速其数学内核;相反,它使用Eigen::TensorcuDNNcuBLASXLA-generated 和手写内核的组合。

    如果您想在 TensorFlow 操作中使用 NNPACK,您可以通过使用该库的 adding a custom op kernel 来执行此操作。

    【讨论】:

      猜你喜欢
      • 2021-04-08
      • 2016-08-03
      • 2016-03-05
      • 2020-01-05
      • 2018-01-15
      • 1970-01-01
      • 1970-01-01
      • 2021-06-04
      • 2018-03-03
      相关资源
      最近更新 更多