Searched refs:is_scalar_batch (Results 1 – 22 of 22) sorted by relevance
61 if self.is_scalar_batch:
67 self._is_scalar_batch = self.bijector.is_scalar_batch
105 if self.is_scalar_batch:
99 if self.is_scalar_batch:
134 if self.is_scalar_batch:
124 def is_scalar_batch(self): member in Bijector
105 self._is_scalar_batch = self.distribution.is_scalar_batch and self.bijector.is_scalar_batch
159 if self.is_scalar_batch:
133 if self.is_scalar_batch:
116 if self.is_scalar_batch:
157 if self.is_scalar_batch:
160 if self.is_scalar_batch:
169 if not self.is_scalar_batch:
168 if self.is_scalar_batch:
177 if self.is_scalar_batch:
185 if self.is_scalar_batch:
184 if self.is_scalar_batch:
174 if self.is_scalar_batch:
175 if self.is_scalar_batch:
131 def is_scalar_batch(self): member in Distribution