Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/parallel_for/
Dtest_util.py33 def _run_targets(self, targets1, targets2=None, run_init=True): argument
35 targets2 = ([] if targets2 is None else nest.flatten(targets2))
36 assert len(targets1) == len(targets2) or not targets2
40 return self.evaluate(targets1 + targets2)
42 def run_and_assert_equal(self, targets1, targets2): argument
43 outputs = self._run_targets(targets1, targets2)
Dgradients_test.py358 def run_and_assert_equal(self, targets1, targets2, atol=1e-4, rtol=1e-4): argument
360 targets2 = nest.flatten(targets2)
361 assert len(targets1) == len(targets2)
364 outputs = self.evaluate(targets1 + targets2)
Dcontrol_flow_ops_test.py414 def run_and_assert_equal(self, targets1, targets2): argument
415 outputs = self._run_targets(targets1, targets2)