Add Operation._remove_all_control_inputs and use in ControlFlowContext.
This allows while loop gradients to work with the C API. This change also enables the C API for control flow tests. PiperOrigin-RevId: 178438424
Loading
Please sign in to comment