tensorflow - 是否有 CPU 支持的 TensorFlow Ops 列表？

Question

我正在尝试在 CPU 上运行为 GPU 编写的模型，并发现该tf.nn.bias_add函数在 CPU 上执行时不支持data_format“NCHW”属性，它仅支持“NHWC”。

是否有一个列表，列出了哪些操作，比如这个，仅限于 GPU？我还没有找到一个。

Answer 1

不，没有这样的列表，遗憾的是文档也没有说明这些细节。

曾尝试在此处要求改进文档，但看起来并没有实现。