Searched refs:_batch_shape_tensor (Results 1 – 16 of 16) sorted by relevance
626 def _batch_shape_tensor(self): member in Distribution647 return self._batch_shape_tensor()
150 def _batch_shape_tensor(self): member in Uniform
109 def _batch_shape_tensor(self): member in Bernoulli
140 def _batch_shape_tensor(self): member in Laplace
230 def _batch_shape_tensor(self): member in Multinomial
244 def _batch_shape_tensor(self): member in DirichletMultinomial
218 def _batch_shape_tensor(self): member in Dirichlet
174 def _batch_shape_tensor(self): member in Normal
198 def _batch_shape_tensor(self): member in Gamma
255 def _batch_shape_tensor(self): member in Categorical
231 def _batch_shape_tensor(self): member in Beta
227 def _batch_shape_tensor(self): member in StudentT
358 def _batch_shape_tensor(self): member in TransformedDistribution
220 shape=self._batch_shape_tensor(shape=shape), dtype=self.dtype)
374 return self._batch_shape_tensor()376 def _batch_shape_tensor(self, shape=None): member in LinearOperator
511 batch_shape = self._batch_shape_tensor(