Searched refs:_possibly_broadcast_batch_shape (Results 1 – 2 of 2) sorted by relevance
259 def _possibly_broadcast_batch_shape(self, x): member in LinearOperatorZeros306 return self._possibly_broadcast_batch_shape(316 return self._possibly_broadcast_batch_shape(zeros)344 return self._possibly_broadcast_batch_shape(mat)
315 def _possibly_broadcast_batch_shape(self, x): member in LinearOperatorIdentity354 return self._possibly_broadcast_batch_shape(x)