Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dtest_util.py32 def _run_targets(self, targets1, targets2=None, run_init=True): argument
33 targets1 = nest.flatten(targets1)
35 assert len(targets1) == len(targets2) or not targets2
39 return self.evaluate(targets1 + targets2)
42 def run_and_assert_equal(self, targets1, targets2, rtol=1e-4, atol=1e-5): argument
43 outputs = self._run_targets(targets1, targets2)
Dgradients_test.py357 def run_and_assert_equal(self, targets1, targets2, atol=1e-4, rtol=1e-4): argument
358 targets1 = nest.flatten(targets1)
360 assert len(targets1) == len(targets2)
363 outputs = self.evaluate(targets1 + targets2)
Dcontrol_flow_ops_test.py667 def run_and_assert_equal(self, targets1, targets2, rtol=1e-4, atol=1e-5): argument
668 outputs = self._run_targets(targets1, targets2)
765 def run_and_assert_equal(self, targets1, targets2, rtol=1e-4, atol=1e-5): argument
766 outputs = self._run_targets(targets1, targets2)