Extend the type info analyzer to cover variables declared using with statements.
This allows constructs of the kind:
with tfe.GradientTape() as tape:
tape.gradients(...)
PiperOrigin-RevId: 181358791
Loading
Please sign in to comment
This allows constructs of the kind:
with tfe.GradientTape() as tape:
tape.gradients(...)
PiperOrigin-RevId: 181358791