tf.math.segment_max/min/mean/sun/prod crashes(aborts) when segment_ids is large #46888
Labels
comp:ops
OPs related issues
stat:awaiting tensorflower
Status - Awaiting response from tensorflower
TF 2.4
for issues related to TF 2.4
type:bug
Bug
System information
Describe the current behavior
tf.math.segment_max/min/mean/sun/prod crashes(aborts) when
segment_idsis largeDescribe the expected behavior
expect an exception message if the input is unexpected instead of crash
Standalone code to reproduce the issue
Output:
Related issue: #46696
The text was updated successfully, but these errors were encountered: