(xs, k=3)
| 330 | def testStopGradients(self): |
| 331 | def _MakeGraph(rng, stop_gradients=()): |
| 332 | def _FunctionOf(xs, k=3): |
| 333 | return ops.convert_to_tensor( |
| 334 | sum(math_ops.matmul(rng.rand(k, k), x) for x in xs) |
| 335 | + rng.rand(k, k)) |
| 336 | |
| 337 | a = _FunctionOf([]) |
| 338 | if "a" in stop_gradients: a = array_ops.stop_gradient(a) |