Searched refs:_get_optimized_reduction_ops_cache_key (Results 1 – 1 of 1) sorted by relevance
1068 cache_key = self._get_optimized_reduction_ops_cache_key(1100 def _get_optimized_reduction_ops_cache_key( member in _WhileBodyGradFuncGraph