Searched refs:_verify_param_shape_across_processes (Results 1 – 3 of 3) sorted by relevance
283 def _verify_param_shape_across_processes( function
40 _verify_param_shape_across_processes,825 _verify_param_shape_across_processes(self.process_group, parameters)
48 _verify_param_shape_across_processes,8432 _verify_param_shape_across_processes(8436 _verify_param_shape_across_processes(