Python initialize_tpu_system Examples

Example #1

File: combinations.py Project: zyc2621854/tensorflow

 def _create_tpu_strategy():
     resolver = cluster_resolver.TPUClusterResolver("")
     tpu_lib.initialize_tpu_system(resolver)
     strategy = tpu_lib.TPUStrategy(resolver,
                                    steps_per_run=steps_per_run,
                                    **kwargs)
     return strategy

Example #2

File: combinations.py Project: jackd/tensorflow

  def _create_tpu_strategy():
    resolver = cluster_resolver.TPUClusterResolver("")
    topology = tpu_lib.initialize_tpu_system(resolver)
    device_assignment = None
    if use_single_core:
      device_assignment = device_assignment_lib.DeviceAssignment(
          topology, core_assignment=device_assignment_lib.
          SINGLE_CORE_ASSIGNMENT)

    strategy = tpu_lib.TPUStrategy(resolver, steps_per_run=steps_per_run,
                                   device_assignment=device_assignment,
                                   **kwargs)
    return strategy

Example #3

File: combinations.py Project: wdma/tensorflow

  def _create_tpu_strategy():
    resolver = cluster_resolver.TPUClusterResolver("")
    topology = tpu_lib.initialize_tpu_system(resolver)
    device_assignment = None
    if use_single_core:
      device_assignment = device_assignment_lib.DeviceAssignment(
          topology, core_assignment=device_assignment_lib.
          SINGLE_CORE_ASSIGNMENT)

    strategy = tpu_lib.TPUStrategy(resolver, steps_per_run=steps_per_run,
                                   device_assignment=device_assignment,
                                   **kwargs)
    return strategy

Example #4

File: combinations.py Project: Wajih-O/tensorflow

 def _create_tpu_strategy():
   resolver = cluster_resolver.TPUClusterResolver("")
   tpu_lib.initialize_tpu_system(resolver)
   strategy = tpu_lib.TPUStrategy(resolver, steps_per_run=steps_per_run,
                                  **kwargs)
   return strategy