Searched refs:pick_vector (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | transformed_distribution.py | 341 distribution_util.pick_vector(
363 return distribution_util.pick_vector(
386 distribution_util.pick_vector(self._needs_rotation, self._empty, [n]),
389 distribution_util.pick_vector(self._needs_rotation, [n], self._empty))
|
| D | distribution.py | 1251 expanded_shape = util.pick_vector(
|
| D | util.py | 660 def pick_vector(cond, true_vector, false_vector, name="pick_vector"): function
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 533 x, self.evaluate(du.pick_vector(math_ops.less(0, 5), x, y)))
535 y, self.evaluate(du.pick_vector(math_ops.less(5, 0), x, y)))
537 du.pick_vector(
540 du.pick_vector(
|