Searched refs:TPUStrategy (Results 1 – 15 of 15) sorted by relevance
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 1 path: "tensorflow.distribute.experimental.TPUStrategy" 3 is_instance: "<class \'tensorflow.python.distribute.tpu_strategy.TPUStrategy\'>"
|
D | tensorflow.distribute.experimental.pbtxt | 16 name: "TPUStrategy"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.distribute.experimental.-t-p-u-strategy.pbtxt | 1 path: "tensorflow.distribute.experimental.TPUStrategy" 3 is_instance: "<class \'tensorflow.python.distribute.tpu_strategy.TPUStrategy\'>"
|
D | tensorflow.distribute.experimental.pbtxt | 16 name: "TPUStrategy"
|
/external/tensorflow/tensorflow/contrib/distribute/python/ |
D | keras_correctness_test_base.py | 127 if isinstance(distribution, tpu_strategy.TPUStrategy): 292 isinstance(distribution, tpu_strategy.TPUStrategy) and 309 isinstance(distribution, tpu_strategy.TPUStrategy)) 468 if (isinstance(distribution, tpu_strategy.TPUStrategy) and
|
D | tpu_strategy.py | 25 from tensorflow.python.distribute.tpu_strategy import TPUStrategy
|
D | checkpointing_test.py | 75 strategy = tpu_strategy.TPUStrategy()
|
D | keras_utils_test.py | 96 if isinstance(distribution, tpu_strategy.TPUStrategy): 517 if (isinstance(distribution, tpu_strategy.TPUStrategy) and
|
D | combinations.py | 340 strategy = tpu_lib.TPUStrategy(resolver, steps_per_run=steps_per_run,
|
D | metrics_v1_test.py | 99 if isinstance(distribution, tpu_strategy.TPUStrategy):
|
D | keras_backward_compat_test.py | 167 if isinstance(distribution, tpu_strategy.TPUStrategy): 1073 isinstance(distribution, tpu_strategy.TPUStrategy) and
|
D | keras_test.py | 201 if isinstance(distribution, tpu_strategy.TPUStrategy):
|
/external/tensorflow/tensorflow/contrib/distribute/ |
D | __init__.py | 34 from tensorflow.contrib.distribute.python.tpu_strategy import TPUStrategy
|
/external/tensorflow/tensorflow/contrib/distribute/python/examples/ |
D | mnist_tf1_tpu.py | 90 strategy = tf.contrib.distribute.TPUStrategy(cluster_resolver)
|
/external/tensorflow/tensorflow/python/distribute/ |
D | tpu_strategy.py | 125 class TPUStrategy(distribute_lib.DistributionStrategy): class 146 super(TPUStrategy, self).__init__(TPUExtended(
|