Searched refs:_possibly_broadcast_batch_shape (Results 1 – 2 of 2) sorted by relevance
277 def _possibly_broadcast_batch_shape(self, x): member in LinearOperatorZeros324 return self._possibly_broadcast_batch_shape(334 return self._possibly_broadcast_batch_shape(zeros)362 return self._possibly_broadcast_batch_shape(mat)
331 def _possibly_broadcast_batch_shape(self, x): member in LinearOperatorIdentity370 return self._possibly_broadcast_batch_shape(x)