Searched refs:pick_vector (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | cholesky_outer_product.py | 195 distribution_util.pick_vector(
223 distribution_util.pick_vector(
|
| /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 | 645 def pick_vector(cond, function
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | conditional_transformed_distribution.py | 57 distribution_util.pick_vector(self._needs_rotation, self._empty, [n]),
60 distribution_util.pick_vector(self._needs_rotation, [n], self._empty))
|
| D | shape.py | 401 event_shape = distribution_util.pick_vector(
404 batch_shape = distribution_util.pick_vector(
|
| D | poisson_lognormal.py | 373 distribution_util.pick_vector(
383 distribution_util.pick_vector(
|
| D | vector_student_t.py | 224 override_batch_shape = distribution_util.pick_vector(
|
| D | mixture_same_family.py | 184 distribution_util.pick_vector(
|
| D | vector_diffeomixture.py | 589 distribution_util.pick_vector(
599 distribution_util.pick_vector(
|
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 532 x, self.evaluate(du.pick_vector(math_ops.less(0, 5), x, y)))
534 y, self.evaluate(du.pick_vector(math_ops.less(5, 0), x, y)))
536 du.pick_vector(
539 du.pick_vector(
|