Commit 88ba64b6 authored by Geoffrey Irving's avatar Geoffrey Irving
Browse files

Two improvements to tf.split's shape function 

1. Don't break if the size corresponding to a -1 is 0.
2. Infer the size corresponding to a -1 if possible.

Fixes #19360.
parent e51791dd
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment