In Tensorflow before version 2.3.1, the RaggedCountSparseOutput
implementation does not validate that the input arguments form a valid ragged tensor. In particular, there is no validation that the values in the splits
tensor generate a valid partitioning of the values
tensor. Hence, the code is prone to heap buffer overflow. If split_values
does not end with a value at least num_values
then the while
loop condition will trigger a read outside of the bounds of split_values
once batch_idx
grows too large. The issue is patched in commit 3cbb917b4714766030b28eba9fb41bb97ce9ee02 and is released in TensorFlow version 2.3.1.