Searched refs:matrix_approximation_rank (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/algorithms/ddp_comm_hooks/ |
| H A D | powerSGD_hook.py | 78 num_rows, num_cols, matrix_approximation_rank, min_compression_rate argument
98 compressed_size = (num_rows + num_cols) * matrix_approximation_rank
182 matrix_approximation_rank=1, argument
196 matrix_approximation_rank,
208 self.matrix_approximation_rank = matrix_approximation_rank
448 matrix_approximation_rank = min(n, m, state.matrix_approximation_rank)
450 n, m, matrix_approximation_rank, state.min_compression_rate
455 total_Ps_size += n * matrix_approximation_rank
456 total_Qs_size += m * matrix_approximation_rank
522 matrix_approximation_rank = min(n, m, state.matrix_approximation_rank)
[all …]
|
| H A D | __init__.py | 27 matrix_approximation_rank, argument
38 matrix_approximation_rank=matrix_approximation_rank,
70 matrix_approximation_rank=1,
77 matrix_approximation_rank=2,
83 matrix_approximation_rank=1,
88 matrix_approximation_rank=2,
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_common.py | 742 matrix_approximation_rank=1,
|
| H A D | test_c10d_nccl.py | 1903 matrix_approximation_rank=1,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 5156 matrix_approximation_rank=1,
7393 matrix_approximation_rank=1,
10313 matrix_approximation_rank=1,
|