generate_local_weight_sharding_params_for_test (reference) in projects: aosp_15_r20

Searched refs:generate_local_weight_sharding_params_for_test (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/ops/
H A D	test_embedding.py	`18 generate_local_weight_sharding_params_for_test, 86 (start_pos, chunk_size) = generate_local_weight_sharding_params_for_test(`
H A D	test_embedding_bag.py	`18 generate_local_weight_sharding_params_for_test, 124 (start_pos, chunk_size) = generate_local_weight_sharding_params_for_test(`
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
H A D	_test_ops_common.py	`80 def generate_local_weight_sharding_params_for_test( function`