Searched refs:recompute_grad (Results 1 – 9 of 9) sorted by relevance
123 model1_re = tf.recompute_grad(model1)124 model2_re = tf.recompute_grad(model2)125 model3_re = tf.recompute_grad(model3)
47 test_model_re = tf.recompute_grad(test_model)
489 def recompute_grad(f): function
1453 test_fn_re = custom_gradient.recompute_grad(test_fn)1499 @custom_gradient.recompute_grad1571 test_fn_re = custom_gradient.recompute_grad(TestFn)
1618 @custom_gradient.recompute_grad1631 @custom_gradient.recompute_grad1647 @custom_gradient.recompute_grad
443 @custom_gradient.recompute_grad
900 name: "recompute_grad"
1888 name: "recompute_grad"
3179 * `tf.contrib.layers.recompute_grad` works for explicit gradient checkpointing on TPU.