Searched refs:grad_pass_through (Results 1 – 5 of 5) sorted by relevance
562 def grad_pass_through(f): function
1636 y = custom_gradient.grad_pass_through(1645 y = custom_gradient.grad_pass_through(lambda v: x * v)(z)1665 y = custom_gradient.grad_pass_through(x.assign)(z**2)1671 y = custom_gradient.grad_pass_through(lambda v: x * v)(z)
696 name: "grad_pass_through"
1416 name: "grad_pass_through"
458 `tf.grad_pass_through(tf.quantization.quantize_and_dequantize_v2)(...)`.