Home
last modified time | relevance | path

Searched refs:tpu_lib (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
DBUILD86 ":tpu_lib",
134 ":tpu_lib",
190 ":tpu_lib",
208 ":tpu_lib",
213 name = "tpu_lib",
448 ":tpu_lib",
466 ":tpu_lib",
490 ":tpu_lib",
504 ":tpu_lib",
566 "//tensorflow/python/tpu:tpu_lib",
[all …]
Dtpu_outside_compilation_test.py30 from tensorflow.python.distribute import tpu_strategy as tpu_lib unknown
76 return tpu_lib.TPUStrategyV2(resolver)
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_compilation_test.py21 from tensorflow.python.distribute import tpu_strategy as tpu_lib unknown
49 strategy = tpu_lib.TPUStrategyV2(resolver)
Dtpu_strategy_test.py31 from tensorflow.python.distribute import tpu_strategy as tpu_lib unknown
84 strategy = tpu_lib.TPUStrategyV2(resolver)
203 strategy = tpu_lib.TPUStrategyV2(
210 strategy2 = tpu_lib.TPUStrategyV2(
311 all_core_strategy = tpu_lib.TPUStrategyV2(resolver)
321 first_core_strategy = tpu_lib.TPUStrategyV2(
329 second_core_strategy = tpu_lib.TPUStrategyV2(
356 first_core_strategy = tpu_lib.TPUStrategyV2(
364 second_core_strategy = tpu_lib.TPUStrategyV2(
976 strategy = tpu_lib.TPUStrategyV2(resolver)
[all …]
Dstrategy_combinations.py32 from tensorflow.python.distribute import tpu_strategy as tpu_lib unknown
112 strategy = tpu_lib.TPUStrategy(resolver, device_assignment, **kwargs)
114 strategy = tpu_lib.TPUStrategyV1(resolver, steps_per_run,
DBUILD610 "//tensorflow/python/tpu:tpu_lib",
900 "//tensorflow/python/tpu:tpu_lib",
1347 "//tensorflow/python/tpu:tpu_lib",
/external/tensorflow/tensorflow/python/compiler/xla/
DBUILD92 "//tensorflow/python/tpu:tpu_lib",
/external/tensorflow/tensorflow/python/keras/utils/
DBUILD183 "//tensorflow/python/tpu:tpu_lib",
/external/tensorflow/tensorflow/python/keras/engine/
DBUILD89 "//tensorflow/python/tpu:tpu_lib",