Commit 2b5c02c2 authored by Thomas O'Malley's avatar Thomas O'Malley Committed by TensorFlower Gardener
Browse files

Change how TensorFlowOpLayer prevents sublayers from making Keras history. 

Instead of marking input tensors as `_keras_history_checked`, create a
call_context object to determine if we are in a sublayer. Only set
`_keras_history_checked` for tensors that don't come from keras inputs.

PiperOrigin-RevId: 233867280
parent 01cf864b
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