tf.random.uniform is optimized out when it shouldn't be #2204
Labels
bug
Unexpected behaviour that should be corrected (type)
tf2.x / tf.keras
Issue could be related to tf2.x where coremltools isn't supported (component)
triaged
Reviewed and examined, release as been assigned if applicable (status)
馃悶Describing the bug
When making a test model that uses a
tf.function
withtf.random.uniform
, it looks liketf.random.uniform
is optimized out of the network. It shouldn't be.Stack Trace
This stack trace appears when exiting the program. It may be related.
To Reproduce
The graphviz output from debug shoes that the random op is not preserved and stdout shows identical outputs for both
predict
callsSystem environment (please complete the following information):
The text was updated successfully, but these errors were encountered: