Searched refs:orig_shape (Results 1 – 4 of 4) sorted by relevance
295 orig_shape = [np.prod(factors[orig_map == d]) for d in range(orig_rank)]300 orig_dense = np.random.uniform(size=orig_shape)310 orig_shape)
50 orig_shape = d.shape55 return np.reshape(diag_list, orig_shape + (d.shape[-1],))
519 orig_shape = d.shape524 return np.reshape(diag_list, orig_shape + (d.shape[-1],))
119 auto orig_shape = ShapeUtil::MakeShape(S32, {2}); in XLA_TEST_F() local