experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix one_hot fuzzer timeout when input is way too long.

Browse Source 
This makes an upper limit on buffer input, any request for more is just silently ignored

PiperOrigin-RevId: 268871697
This commit is contained in:
Mihai Maruseac 2019-09-13 03:16:18 -07:00 committed by TensorFlower Gardener
parent 4a09233096
commit 1c62eaec25
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/fuzzing/one_hot_fuzz.cc
View File

@ -20,6 +20,9 @@ limitations under the License.
namespace tensorflow {
namespace fuzzing {
// Don't generate tensors that are too large as we don't test that branch here
constexpr size_t kMaxSize = 1024;
class FuzzOneHot : public FuzzSession {
void BuildGraph(const Scope& scope) override {
auto input =
@ -39,6 +42,11 @@ class FuzzOneHot : public FuzzSession {
const uint8_t* input_data;
if (size > 3) {
// Since we only care about the one hot decoding and not about the size of
// the tensor, limit `size` to at most `kMaxSize`.
if (size > kMaxSize) {
size = kMaxSize;
}
depth = static_cast<int32>(data[0]);
on = data[1];
off = data[2];