| /external/tensorflow/tensorflow/python/distribute/ |
| D | mirrored_run.py | 227 merge_result = threads[0].merge_fn(distribution, *merge_args,
261 self.merge_fn = None
391 t.merge_fn = fn
|
| D | distribute_lib.py | 2419 def merge_fn(_, flat_value): function
2423 reduced = replica_context.merge_call(merge_fn, args=(nest.flatten(value),))
2952 def merge_call(self, merge_fn, args=(), kwargs=None): argument
2983 merge_fn = autograph.tf_convert(
2984 merge_fn, autograph_ctx.control_status_ctx(), convert_by_default=False)
2985 return self._merge_call(merge_fn, args, kwargs)
2987 def _merge_call(self, merge_fn, args, kwargs): argument
2992 return merge_fn(self._strategy, *args, **kwargs)
|
| D | distribute_lib_test.py | 445 def merge_fn(dist, s): function
456 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
462 def merge_fn(_, s): function
469 replica_ctx.merge_call(merge_fn, args=("bar",))
|
| D | ps_values.py | 111 def merge_fn(strategy, function
129 return replica_context.merge_call(merge_fn, args=args, kwargs=kwargs)
|
| D | mirrored_strategy_test.py | 430 def merge_fn(strategy, value): function
437 return ds_context.get_replica_context().merge_call(merge_fn, args=(i,))
448 def merge_fn(strategy, value): function
456 return ds_context.get_replica_context().merge_call(merge_fn, args=(t,))
466 def merge_fn(_): function
471 ds_context.get_replica_context().merge_call(merge_fn)
|
| D | input_lib.py | 2074 def merge_fn(distribution, value): function
2083 merge_fn, args=(output,))
2095 def merge_fn(distribution, value): function
2101 merge_fn, args=(output,))
|
| D | values.py | 51 def merge_fn(strategy, value, **kwargs): function
73 merge_fn, args=(value,), kwargs=kwargs)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 32 …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 | 36 …argspec: "args=[\'self\', \'merge_fn\', \'args\', \'kwargs\'], varargs=None, keywords=None, defaul…
|
| /external/tensorflow/tensorflow/python/training/ |
| D | moving_averages.py | 106 def merge_fn(strategy, v, value): function
111 return replica_context.merge_call(merge_fn, args=(variable, value))
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 129 def merge_fn_wrapper(distribution, merge_fn, *args): argument
132 result = distribution.experimental_local_results(merge_fn)[0](*args)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 85 def merge_lists(repeated1, repeated2, empty_fn, key_fn, merge_fn): argument
93 merged[name] = merge_fn(x1, x2)
|