Searched refs:_compare_tensor (Results 1 – 2 of 2) sorted by relevance
23 def _compare_tensor(self, orig_tensor, dist_tensor, offload_to_cpu=False): member in VerifyStateDictMixin47 self._compare_tensor(param, dist_param, offload_to_cpu)80 self._compare_tensor(state, dist_state)
378 self._compare_tensor(