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
248 def _batch_shape_tensor(self): member in DirichletMultinomial
144 def _batch_shape_tensor(self): member in Laplace
234 def _batch_shape_tensor(self): member in Multinomial
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
235 def _batch_shape_tensor(self): member in Beta
633 def _batch_shape_tensor(self): member in Distribution654 return self._batch_shape_tensor()
231 def _batch_shape_tensor(self): member in StudentT
362 def _batch_shape_tensor(self): member in TransformedDistribution
225 shape=self._batch_shape_tensor(shape=shape), dtype=self.dtype)
370 return self._batch_shape_tensor()372 def _batch_shape_tensor(self, shape=None): member in LinearOperator
388 batch_shape = self._batch_shape_tensor(