Searched refs:_maybe_partial_apply_variables (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | tpu_strategy.py | 108 def _maybe_partial_apply_variables(fn, args, kwargs): function
396 fn, args, kwargs = _maybe_partial_apply_variables(fn, args, kwargs)
659 fn, args, kwargs = _maybe_partial_apply_variables(fn, args, kwargs)
775 fn, args, kwargs = _maybe_partial_apply_variables(fn, args, kwargs)
|