Ejemplos de initialize_tpu_system en Python

Ejemplo n.º 1

Archivo: combinations.py Proyecto: 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

Ejemplo n.º 2

Archivo: combinations.py Proyecto: 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

Ejemplo n.º 3

Archivo: combinations.py Proyecto: 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

Ejemplo n.º 4

Archivo: combinations.py Proyecto: 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