We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
DNNL MaxPool should get nan value when input values are all nan.
test case:
import torch import intel_pytorch_extension from torch._six import inf, nan intel_pytorch_extension.core.enable_auto_dnnl() input = torch.full([1, 1] + 2 * [3], nan).to('dpcpp') output = F.max_pool2d(x, 3, 3)