def my_add_control_inputs()

in memory_saving_gradients.py [0:0]

• 4 lines of code
• 5 McCabe index (conditional complexity)

def my_add_control_inputs(wait_to_do_ops, inputs_to_do_before):
    for op in wait_to_do_ops:
        ci = [i for i in inputs_to_do_before if op.control_inputs is None or i not in op.control_inputs]
        ge.add_control_inputs(op, ci)