Searched refs:_dense_apply_args (Results 1 – 1 of 1) sorted by relevance
694 if "apply_state" in self._dense_apply_args:1314 def _dense_apply_args(self): member in OptimizerV2