标签: tensorflow tensorflow-gpu
我正在尝试运行一个在CPU上为GPU编写的模型,并发现tf.nn.bias_add函数不支持" NCHW"的data_format属性。在CPU上执行时,它只支持" NHWC"。
tf.nn.bias_add
data_format
是否列出了哪些操作(如此操作)仅限于GPU?我还没能找到一个。
答案 0 :(得分:0)
不,没有这样的清单,遗憾的是文档也没有说明这些细节。
试图要求提供文档改进here,但看起来并没有实现。