Searched defs:replica_fn (Results 1 – 12 of 12) sorted by relevance
233 def replica_fn(): function274 def replica_fn(): function649 def replica_fn(): function717 def replica_fn(): function790 def replica_fn(): function850 def replica_fn(): function891 def replica_fn(): function938 def replica_fn(): function1009 def replica_fn(): function1065 def replica_fn(): function[all …]
76 def replica_fn(): function282 def replica_fn(): function327 def replica_fn(): function345 def replica_fn(): function371 def replica_fn(): function425 def replica_fn(): function473 def replica_fn(): function491 def replica_fn(): function520 def replica_fn(): function561 def replica_fn(): function
56 def replica_fn(): function83 def replica_fn(): function166 def replica_fn(): function
363 def replica_fn(): function396 def replica_fn(): function
275 def replica_fn(per_replica_value): function347 def replica_fn(per_replica_value): function519 def replica_fn(value): function
66 def run_on_strategy(replica_fn, strat, coord): argument
120 def replica_fn(input_tensor): function1412 def replica_fn(): function1447 def replica_fn(): function
718 def replica_fn(): function
543 def replica_fn(): function
180 def replica_fn(elem): function286 def replica_fn(elem): function341 def replica_fn(elem): function
1062 def replica_fn(input_tensor): function1141 def replica_fn(input_tensor): function1178 def replica_fn(input_tensor): function1214 def replica_fn(): function1261 def replica_fn(): function1282 def replica_fn(): function
203 def replica_fn(): function