Searched refs:_broadcast_to_shape (Results 1 – 1 of 1) sorted by relevance
200 batch_matrices[i] = _broadcast_to_shape(mat, bcast_shape)211 batch_matrices[i] = _broadcast_to_shape(mat, bcast_shape)216 def _broadcast_to_shape(x, shape): function