Searched refs:run_as_function_for_tape_gradients (Results 1 – 3 of 3) sorted by relevance
344 def run_as_function_for_tape_gradients(make_op, inputs): function
296 tensors = util.run_as_function_for_tape_gradients(_make_op, cond_inputs)1230 tensors = util.run_as_function_for_tape_gradients(_make_op, case_inputs)
508 return util.run_as_function_for_tape_gradients(_make_op, loop_vars)