Aborted (core dumped) in `tf.irfftnd/ifftnd/irfftnd`

[x0w3n]

Issue type

Bug

Have you reproduced the bug with TensorFlow Nightly?

No

Source

source

TensorFlow version

tf 2.16

Custom code

Yes

OS platform and distribution

Linux Ubuntu 22.04.3 LTS (x86_64)

Mobile device

No response

Python version

3.9.13

Bazel version

No response

GCC/compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current behavior?

When the fft_length parameter of tf.irfftnd/ifftnd/irfftnd is less than or equal to 0, it will trigger a crash, resulting in Check failed.

Standalone code to reproduce the issue

import tensorflow as tf

input1 = tf.random.normal([10,10,10,10,10], dtype=tf.float32)
input2 = tf.random.normal([10,10,10,10,10], dtype=tf.float32)
input3 = tf.complex(input1,input2)

tf.irfftnd(input3,-3,-3) #crash
# tf.ifftnd(input3,-3,-3) #crash
# tf.irfftnd(input3,-3,-3) #crash

Relevant log output

2024-05-07 07:46:42.396021: F tensorflow/core/framework/tensor_shape.cc:45] Check failed: NDIMS == dims() (1 vs. 0)Asking for tensor of 1 dimensions from a tensor of 0 dimensions
Aborted (core dumped)