| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib.py | 2495 def merge_fn(_, flat_value): function
2499 reduced = replica_context.merge_call(merge_fn, args=(nest.flatten(value),))
2511 def merge_fn(_, *merged_args, **merged_kwargs): function
2514 return replica_context.merge_call(merge_fn, args=args, kwargs=kwargs)
3073 def merge_call(self, merge_fn, args=(), kwargs=None): argument
3104 merge_fn = autograph.tf_convert(
3105 merge_fn, autograph_ctx.control_status_ctx(), convert_by_default=False)
3106 return self._merge_call(merge_fn, args, kwargs)
3108 def _merge_call(self, merge_fn, args, kwargs): argument
3113 return merge_fn(self._strategy, *args, **kwargs)
|
| D | mirrored_run.py | 276 merge_result = threads[0].merge_fn(distribution, *merge_args,
310 self.merge_fn = None
459 t.merge_fn = fn
|
| D | distribute_lib_test.py | 462 def merge_fn(dist, s): function
473 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
479 def merge_fn(_, s): function
486 replica_ctx.merge_call(merge_fn, args=("bar",))
|
| D | mirrored_strategy_test.py | 454 def merge_fn(strategy, value): function
461 return ds_context.get_replica_context().merge_call(merge_fn, args=(i,))
472 def merge_fn(strategy, value): function
480 return ds_context.get_replica_context().merge_call(merge_fn, args=(t,))
490 def merge_fn(_): function
495 ds_context.get_replica_context().merge_call(merge_fn)
|
| D | input_lib.py | 2026 def merge_fn(distribution, value): function
2035 merge_fn, args=(output,))
2047 def merge_fn(distribution, value): function
2053 merge_fn, args=(output,))
|
| D | ps_values.py | 130 def merge_fn(strategy, function
148 return replica_context.merge_call(merge_fn, args=args, kwargs=kwargs)
|
| D | values.py | 74 def merge_fn(strategy, value, **kwargs): function
96 merge_fn, args=(value,), kwargs=kwargs)
110 def merge_fn(strategy, value): function
115 return ds_context.get_replica_context().merge_call(merge_fn, args=(value,))
|
| D | strategy_common_test.py | 71 def merge_fn(unused_strat): function
79 y = ds_context.get_replica_context().merge_call(merge_fn)
|
| /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 | 103 def merge_fn(strategy, v, value): function
108 return replica_context.merge_call(merge_fn, args=(variable, value))
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | metrics_utils.py | 173 def merge_fn_wrapper(distribution, merge_fn, *args): argument
176 result = distribution.experimental_local_results(merge_fn)[0](*args)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | convert_to_constants_test.py | 87 def merge_lists(repeated1, repeated2, empty_fn, key_fn, merge_fn): argument
95 merged[name] = merge_fn(x1, x2)
|