Only start_step/end_step on GradientTape if executing eagerly.
This prevents creating a context where none is required. PiperOrigin-RevId: 213500408
Loading
Please sign in to comment
This prevents creating a context where none is required. PiperOrigin-RevId: 213500408