Searched refs:distribute_ctx (Results 1 – 3 of 3) sorted by relevance
27 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown541 if distribute_ctx.has_strategy():543 if distribute_ctx.in_cross_replica_context():548 return distribute_ctx.get_replica_context().merge_call(800 distribution_strategy = distribute_ctx.get_strategy()
27 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown637 replica_context = distribute_ctx.get_replica_context()838 num_replicas = distribute_ctx.get_strategy().num_replicas_in_sync877 return distribute_ctx.get_replica_context().merge_call(
28 from tensorflow.python.distribute import distribution_strategy_context as distribute_ctx unknown410 return distribute_ctx.get_replica_context().merge_call(541 strategy = distribute_ctx.get_strategy()