Searched refs:_batch_shape_tensor (Results 1 – 16 of 16) sorted by relevance
113 def _batch_shape_tensor(self): member in Bernoulli
154 def _batch_shape_tensor(self): member in Uniform
144 def _batch_shape_tensor(self): member in Laplace
234 def _batch_shape_tensor(self): member in Multinomial
248 def _batch_shape_tensor(self): member in DirichletMultinomial
222 def _batch_shape_tensor(self): member in Dirichlet
178 def _batch_shape_tensor(self): member in Normal
202 def _batch_shape_tensor(self): member in Gamma
259 def _batch_shape_tensor(self): member in Categorical
633 def _batch_shape_tensor(self): member in Distribution654 return self._batch_shape_tensor()
235 def _batch_shape_tensor(self): member in Beta
231 def _batch_shape_tensor(self): member in StudentT
362 def _batch_shape_tensor(self): member in TransformedDistribution
226 shape=self._batch_shape_tensor(shape=shape), dtype=self.dtype)
377 return self._batch_shape_tensor()379 def _batch_shape_tensor(self, shape=None): member in LinearOperator
388 batch_shape = self._batch_shape_tensor(