tree e09a4d894fda5a8f03f0e3efc39a29226cbfc8cc
parent 831c4e95cbcde1aa3ae751c55e9fbda31609bb7a
author Mihai Maruseac <mihaimaruseac@google.com> 1637295414 -0800
committer TensorFlower Gardener <gardener@tensorflow.org> 1637295695 -0800

Check for tensors to be vectors in `BoostedTreesSparseAggregateStatsOp`.

Calling `tensor->vec` should only happen after checking that the tensor shape implies a vector. Otherwise, we can get denial of service via `CHECK`-fails

PiperOrigin-RevId: 410960878
Change-Id: I7b26bec796cbaebde4696862eb855160402b4b0d