Commit 28807c56 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

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
parent b1c7d177
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment