Searched refs:grad_pass_through (Results 1 – 5 of 5) sorted by relevance
755 def grad_pass_through(f): function
1690 y = custom_gradient.grad_pass_through(1699 y = custom_gradient.grad_pass_through(lambda v: x * v)(z)1719 y = custom_gradient.grad_pass_through(x.assign)(z**2)1725 y = custom_gradient.grad_pass_through(lambda v: x * v)(z)
704 name: "grad_pass_through"
1424 name: "grad_pass_through"
4586 `tf.grad_pass_through(tf.quantization.quantize_and_dequantize_v2)(...)`.