Searched refs:_calc_batch_shape (Results 1 – 2 of 2) sorted by relevance
82 self._batch_shape = self._calc_batch_shape()185 def _calc_batch_shape(self): member in Bijector
82 self._batch_shape = self._calc_batch_shape()229 def _calc_batch_shape(self): member in Distribution