Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/distributed/
H A Dutils.py283 def _verify_param_shape_across_processes( function
/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Ddistributed.py40 _verify_param_shape_across_processes,
825 _verify_param_shape_across_processes(self.process_group, parameters)
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py48 _verify_param_shape_across_processes,
8432 _verify_param_shape_across_processes(
8436 _verify_param_shape_across_processes(