def model(t):
     with tape_lib.GradientTape() as tape:
         tape.watch(t)
         result = unified_nn_ops.relu(t)
     grads = tape.gradient(result, t)
     return grads
 def model(t):
     return unified_nn_ops.relu(t)