是否有CPU支持的TensorFlow Ops列表?

时间:2017-06-09 21:30:38

标签: tensorflow tensorflow-gpu

我正在尝试运行一个在CPU上为GPU编写的模型,并发现tf.nn.bias_add函数不支持" NCHW"的data_format属性。在CPU上执行时,它只支持" NHWC"。

是否列出了哪些操作(如此操作)仅限于GPU?我还没能找到一个。

1 个答案:

答案 0 :(得分:0)

不,没有这样的清单,遗憾的是文档也没有说明这些细节。

试图要求提供文档改进here,但看起来并没有实现。