Home
last modified time | relevance | path

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 Dtest_multiarray.py5140 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)