Python initialize_tpu_system示例

示例#1

文件： combinations.py 项目： 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

示例#2

文件： combinations.py 项目： 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

示例#3

文件： combinations.py 项目： 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

示例#4

文件： combinations.py 项目： 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