tensorflow 在同一个线程中,对于NCHW和NHWC的tf.nn.max_pool2d输出不正确,

rsl1atfo  于 4个月前  发布在  其他
关注(0)|答案(4)|浏览(41)

问题类型

Bug

你是否在TensorFlow Nightly版本中复现了这个bug?

是的

问题来源

二进制文件

TensorFlow版本

2.13

自定义代码

OS平台和发行版

Ubuntu 22.04

移动设备

  • 无响应*

Python版本

3.10

Bazel版本

  • 无响应*

GCC/编译器版本

  • 无响应*

CUDA/cuDNN版本

  • 无响应*

GPU型号和内存大小

  • 无响应*

当前行为?

当使用相同的批处理大小、高度、宽度和深度,首先以NCHW格式输入,然后以NHWC格式(支持MKL)输入时,tf.nn.max_pool2d的输出不正确。tf.nn.max_pool2d在相同输入(除了数据格式)上的输出必须相同(除了数据格式)。

独立代码重现问题

使用Colab重现 https://colab.research.google.com/drive/1Vuo7txDPDq-YXAuCFQcPNowHgOwbT7Oi?usp=sharing

cl25kdpy

cl25kdpy1#

你好,@API92 ,

感谢你报告这个问题。我已经在intel-tensorflow中复现了报告的错误,并在这里附上了gist

我已经用TF构建的二进制文件交叉检查了相同的代码,它运行正常。为了参考,我附上了gist

这证实了问题仅限于intel-tensorflow。因此,我会向@TensorFlow-MKL团队留言。

感谢你的PR!

d5vmydt9

d5vmydt92#

我们会尽快检查并反馈!

zf9nrax1

zf9nrax13#

这个问题只出现在 TF 2.13 版本中。
请使用 TF 2.12 版本。
我们会尽快向开发团队报告并修复它。
谢谢!

5t7ly7z5

5t7ly7z54#

修复已在#62819中,并应在下一个TF发布版本中提供。

相关问题