Update the tf.constant python fuzzer to import from the tensorflow pip package.

PiperOrigin-RevId: 351645671
Change-Id: I13bf6ce834397a60e5236a4a012ca31c2010d55e
This commit is contained in:
Amit Patankar 2021-01-13 12:47:55 -08:00 committed by TensorFlower Gardener
parent 100b443d8f
commit 5b94f9954b
2 changed files with 2 additions and 5 deletions
tensorflow/security/fuzzing

View File

@ -133,7 +133,4 @@ tf_py_fuzz_target(
name = "constant_fuzz",
srcs = ["constant_fuzz.py"],
tags = ["notap"], # Run in OSS only.
deps = [
"//tensorflow/python:constant_op",
],
)

View File

@ -15,11 +15,11 @@
"""This is a Python API fuzzer for tf.constant."""
import sys
import atheris_no_libfuzzer as atheris
from tensorflow.python.framework import constant_op
import tensorflow as tf
def TestOneInput(data):
constant_op.constant(data)
tf.constant(data)
def main():