Home
last modified time | relevance | path

Searched defs:call_for_each_replica (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dmirrored_run.py42 def call_for_each_replica(strategy, fn, args=None, kwargs=None): function
Ddistribute_lib.py2726 def call_for_each_replica(self, fn, args=(), kwargs=None): member in StrategyExtendedV1