Searched refs:get_alignment_size (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | pad_mm.py | 63 def get_alignment_size(x: Tensor) -> int: function
196 k_padded_length = get_padded_length(mat1.shape[1], get_alignment_size(mat1))
197 n_padded_length = get_padded_length(mat2.shape[1], get_alignment_size(mat2))
198 m_padded_length = get_padded_length(mat1.shape[0], get_alignment_size(mat1))
382 k_padded_length = get_padded_length(k, get_alignment_size(mat1))
383 n_padded_length = get_padded_length(n, get_alignment_size(mat2))
384 m_padded_length = get_padded_length(m, get_alignment_size(mat1))
390 k_padded_length = get_padded_length(k, get_alignment_size(mat1))
391 m_padded_length = get_padded_length(m, get_alignment_size(mat1))
392 n_padded_length = get_padded_length(n, get_alignment_size(mat2))
[all …]
|
| /external/pytorch/test/inductor/ |
| D | test_pad_mm.py | 8 get_alignment_size,
45 aligned_k = get_padded_length(K2, get_alignment_size(a)) + K2
81 aligned_k = get_padded_length(K2, get_alignment_size(a)) + K2
107 aligned_k = get_padded_length(K, get_alignment_size(a)) + K
136 aligned_m = get_padded_length(M, get_alignment_size(a)) + M
202 aligned_k = get_padded_length(K, get_alignment_size(a)) + K
231 aligned_n = get_padded_length(N, get_alignment_size(b)) + N
260 aligned_n = get_padded_length(N, get_alignment_size(b)) + N
290 aligned_k = get_padded_length(K, get_alignment_size(b)) + K
379 expected_alignment = get_alignment_size(mat1)
|