Home
last modified time | relevance | path

Searched refs:merge_call (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py31 def merge_call(self, fn, *args, **kwargs): member in _TestReplicaContext
84 self.assertEqual("foo", replica_context.merge_call(None, test_arg="foo"))
145 self.assertEqual("foo_bar", replica_ctx.merge_call(merge_fn, args=("bar",)))
Ddistribute_lib.py1410 def merge_call(self, merge_fn, args=(), kwargs=None): member in ReplicaContext
1509 ys = self.merge_call(batch_all_reduce, args=xs)
1516 value, self.merge_call(batch_all_reduce, args=nest.flatten(value)))
Dinput_lib.py635 distribution_strategy_context.get_replica_context().merge_call(
653 distribution_strategy_context.get_replica_context().merge_call(
Dvalues.py695 return distribution_strategy_context.get_replica_context().merge_call(
969 return distribution_strategy_context.get_replica_context().merge_call(
1506 return replica_context.merge_call(merge_fn, args=args, kwargs=kwargs)
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy_multigpu_test.py240 ds_context.get_replica_context().merge_call(lambda _: _)
299 ds_context.get_replica_context().merge_call(merge_fn)
327 ds_context.get_replica_context().merge_call(lambda _: _)
343 ds_context.get_replica_context().merge_call(lambda _: _)
353 ds_context.get_replica_context().merge_call(lambda _: _)
365 ds_context.get_replica_context().merge_call(lambda _: _)
380 ds_context.get_replica_context().merge_call(lambda _: _)
397 ds_context.get_replica_context().merge_call(lambda _: _)
414 ds_context.get_replica_context().merge_call(lambda _: _)
442 ds_context.get_replica_context().merge_call(lambda _: _)
[all …]
Dstrategy_test_lib.py66 ds_context.get_replica_context().merge_call(_raise_exception_fn)
80 ds_context.get_replica_context().merge_call(_call_raises_fn)
95 ds_context.get_replica_context().merge_call(_call_merge_raises_fn)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-replica-context.pbtxt30 name: "merge_call"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-replica-context.pbtxt30 name: "merge_call"
/external/tensorflow/tensorflow/python/kernel_tests/
Dtemplate_mirrored_strategy_test.py37 ds_context.get_replica_context().merge_call(lambda _: ())
/external/tensorflow/tensorflow/contrib/metrics/python/metrics/
Dclassification.py177 best_f1 = distribution_strategy_context.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py191 distribution_strategy_context.get_replica_context().merge_call(
Dmoving_averages.py103 return replica_context.merge_call(merge_fn, args=(variable, value))
Doptimizer.py548 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/keras/utils/
Dmetrics_utils.py122 result_t = replica_context.merge_call(
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Doptimizer_v2.py646 self._per_graph_state = replica_context.merge_call(lambda _: {})
877 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py410 return distribute_ctx.get_replica_context().merge_call(
/external/tensorflow/tensorflow/python/ops/
Dmetrics_impl.py311 return distribution_strategy_context.get_replica_context().merge_call(