Commit 40d5bf33 authored by Skye Wanderman-Milne's avatar Skye Wanderman-Milne Committed by TensorFlower Gardener
Browse files

Enable Operation._add_control_inputs() with the C API and related improvements

This change:
 - Implements the C API logic for Operation._add_control_inputs()
 - Adds type-checking to Operation._add_control_input()
 - Makes Graph::AddControlEdge() update the node def if necessary
 - Makes Graph::AddControlEdge() a no-op if the control edge already exists

The AddControlEdge() changes may have a performance impact if anything
is sensitive to AddControlEdge(), but nothing is to my knowledge. I'm
not sure what benchmarks would confirm this.

PiperOrigin-RevId: 172158589
parent 7679a2ec
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment