Searched refs:aligned_array (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/torch_np/numpy_tests/core/ |
| H A D | test_multiarray.py | 5140 def aligned_array(shape, align, dtype, order="C"): function
5152 aligned = aligned_array(arr.shape, align, dtype, order)
5159 m = aligned_array(100, 15, np.float32)
5160 s = aligned_array((100, 100), 15, np.float32)
|