Home
last modified time | relevance | path

Searched refs:p_iters (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py2606 def _testWhileGrad_Mul(self, use_gpu, p_iters): argument
2612 r = control_flow_ops.while_loop(c, b, [v], parallel_iterations=p_iters)
2621 self._testWhileGrad_Mul(use_gpu=False, p_iters=1)
2622 self._testWhileGrad_Mul(use_gpu=False, p_iters=10)
2623 self._testWhileGrad_Mul(use_gpu=True, p_iters=1)
2624 self._testWhileGrad_Mul(use_gpu=True, p_iters=10)