| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 628 def merge_fn(distribution, value): function
636 merge_fn, args=(output,))
648 def merge_fn(distribution, value): function
654 merge_fn, args=(output,))
|
| D | distribute_lib_test.py | 134 def merge_fn(dist, s): function
145 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
|
| D | distribute_lib.py | 1410 def merge_call(self, merge_fn, args=(), kwargs=None): argument
1442 return self._merge_call(merge_fn, args, kwargs)
1444 def _merge_call(self, merge_fn, args, kwargs): argument
1449 return merge_fn(self._strategy, *args, **kwargs)
|
| D | values.py | 690 def merge_fn(strategy, value, *other_args, **other_kwargs): function
696 merge_fn, args=args, kwargs=kwargs)
964 def merge_fn(strategy, value, *other_args, **other_kwargs): function
970 merge_fn, args=args, kwargs=kwargs)
1501 def merge_fn(strategy, value, *other_args, **other_kwargs): function
1506 return replica_context.merge_call(merge_fn, args=args, kwargs=kwargs)
|
| D | mirrored_strategy.py | 186 merge_result = threads[0].merge_fn(distribution, *merge_args,
821 self.merge_fn = None
921 t.merge_fn = fn
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 31 …argspec: "args=[\'self\', \'merge_fn\', \'args\', \'kwargs\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 31 …argspec: "args=[\'self\', \'merge_fn\', \'args\', \'kwargs\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 110 def merge_fn_wrapper(distribution, merge_fn, *args): argument
113 merged_result_fn = distribution.unwrap(merge_fn)[0](*args)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 98 def merge_fn(strategy, v, value): function
103 return replica_context.merge_call(merge_fn, args=(variable, value))
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | mirrored_strategy_multigpu_test.py | 294 def merge_fn(_): function
299 ds_context.get_replica_context().merge_call(merge_fn)
|