Crash in TensorFlow when using API `tf.raw_ops.DeleteMultiDeviceIterator`

Issue type

Bug

Have you reproduced the bug with TensorFlow Nightly?

Yes

Source

source

TensorFlow version

tf-2.19

Custom code

Yes

OS platform and distribution

No response

Mobile device

No response

Python version

No response

Bazel version

No response

GCC/compiler version

No response

CUDA/cuDNN version

No response

GPU model and memory

No response

Current behavior?

I encountered a crash in TensorFlow when I used API tf.raw_ops.DeleteMultiDeviceIterator. The code is as follows:

import tensorflow as tf

handle = tf.raw_ops.MultiDeviceIterator(
    devices=["/device:CPU:0"],
    shared_name="test_iterator",
    container="",
    output_types=[tf.float32],
    output_shapes=[tf.TensorShape([None, 10])]
)

deleter = tf.raw_ops.DeleteMultiDeviceIterator(
    multi_device_iterator=handle,
    iterators=[
        tf.raw_ops.MultiDeviceIteratorGetNextFromShard(
            multi_device_iterator=handle,
            shard_num=0,
            incarnation_id=0,
            output_types=[tf.float32],
            output_shapes=[tf.TensorShape([None, 10])]
        )
    ],
    deleter=tf.raw_ops.DeleteMultiDeviceIterator
)

When running the above code, TensorFlow crashes instead of reporting an invalid argument error.

[1] 3755469 segmentation fault (core dumped)

The above code would crash on TensorFlow nightly builds (2.19). The gist can be found here.

Standalone code to reproduce the issue

import tensorflow as tf

handle = tf.raw_ops.MultiDeviceIterator(
    devices=["/device:CPU:0"],
    shared_name="test_iterator",
    container="",
    output_types=[tf.float32],
    output_shapes=[tf.TensorShape([None, 10])]
)

deleter = tf.raw_ops.DeleteMultiDeviceIterator(
    multi_device_iterator=handle,
    iterators=[
        tf.raw_ops.MultiDeviceIteratorGetNextFromShard(
            multi_device_iterator=handle,
            shard_num=0,
            incarnation_id=0,
            output_types=[tf.float32],
            output_shapes=[tf.TensorShape([None, 10])]
        )
    ],
    deleter=tf.raw_ops.DeleteMultiDeviceIterator
)

Relevant log output

> [1]    3755469 segmentation fault (core dumped)