Searched refs:distribute_ctx (Results 1 – 4 of 4) sorted by relevance
22 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown45 reduced = distribute_ctx.get_replica_context().merge_call(94 if isinstance(distribute_ctx.get_strategy(),115 if isinstance(distribute_ctx.get_strategy(),136 if isinstance(distribute_ctx.get_strategy(),
29 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown405 if distribute_ctx.has_strategy():406 self._distribution_strategy = distribute_ctx.get_strategy()646 if distribute_ctx.in_cross_replica_context():652 strategy = distribute_ctx.get_strategy()669 return distribute_ctx.get_replica_context().merge_call(897 strategy = distribute_ctx.get_strategy()1409 if self._distribution_strategy and not distribute_ctx.has_strategy():
29 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown533 num_replicas = distribute_ctx.get_strategy().num_replicas_in_sync569 if distribute_ctx.has_strategy():571 if distribute_ctx.in_cross_replica_context():576 return distribute_ctx.get_replica_context().merge_call(821 distribution_strategy = distribute_ctx.get_strategy()
32 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown296 if distribute_ctx.has_strategy():297 strategy = distribute_ctx.get_strategy()