Home
last modified time | relevance | path

Searched refs:tpu_strategy (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_type_spec_test.py33 from tensorflow.python.distribute import tpu_strategy
57 strategy_combinations.tpu_strategy,
89 strategy_combinations.tpu_strategy,
122 strategy_combinations.tpu_strategy,
168 strategy_combinations.tpu_strategy,
208 strategy_combinations.tpu_strategy,
232 strategy_combinations.tpu_strategy,
399 strategy_combinations.tpu_strategy,
424 (tpu_strategy.TPUStrategyV2, tpu_strategy.TPUStrategy)):
452 strategy_combinations.tpu_strategy,
[all …]
Dstrategy_common_test.py31 from tensorflow.python.distribute import tpu_strategy
119 strategy_combinations.tpu_strategy,
126 if (isinstance(strategy, tpu_strategy.TPUStrategy) and
147 if (isinstance(strategy, tpu_strategy.TPUStrategy) and
184 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1,
185 tpu_strategy.TPUStrategyV2))
321 strategy, tpu_strategy.TPUStrategy):
Dmoving_averages_test.py27 from tensorflow.python.distribute import tpu_strategy
40 strategy_combinations.tpu_strategy,
194 distribution, (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
256 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
DREADME.md52 tpu_strategy = tf.distribute.TPUStrategy(resolver)
56 with tpu_strategy.scope():
70 return tpu_strategy.run(
75 input_iterator = iter(tpu_strategy.experimental_distribute_dataset(dataset))
Dstrategy_combinations.py32 from tensorflow.python.distribute import tpu_strategy as tpu_lib
231 tpu_strategy = combinations.NamedDistribution( variable
349 tpu_strategy, # steps_per_run=2
364 tpu_strategy, # steps_per_run=2
379 tpu_strategy, # steps_per_run=2
Dstrategy_combinations_test.py32 from tensorflow.python.distribute import tpu_strategy
138 self.assertIsInstance(strategy, tpu_strategy.TPUStrategyV1)
191 self.assertIsInstance(strategy, tpu_strategy.TPUStrategy)
Dmetrics_v1_test.py24 from tensorflow.python.distribute import tpu_strategy
90 strategy_combinations.tpu_strategy
102 if isinstance(distribution, (tpu_strategy.TPUStrategy,
103 tpu_strategy.TPUStrategyV1)):
Dvalues_test.py38 from tensorflow.python.distribute import tpu_strategy
109 strategy_combinations.tpu_strategy,
281 strategy_combinations.tpu_strategy,
302 strategy_combinations.tpu_strategy,
402 strategy_combinations.tpu_strategy,
488 tpu_strategy.TPUExtended) and context.executing_eagerly():
502 if isinstance(distribution.extended, tpu_strategy.TPUExtended):
562 if not isinstance(distribution.extended, tpu_strategy.TPUExtended):
590 if not (isinstance(distribution.extended, tpu_strategy.TPUExtended) and
820 if isinstance(distribution.extended, tpu_strategy.TPUExtended):
[all …]
Dstrategy_gather_test.py30 from tensorflow.python.distribute import tpu_strategy
63 strategy_combinations.tpu_strategy,
676 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1,
677 tpu_strategy.TPUStrategyV2))
Dcheckpointing_test.py39 strategy_combinations.tpu_strategy,
Dtpu_strategy_compilation_test.py21 from tensorflow.python.distribute import tpu_strategy as tpu_lib
DBUILD578 name = "tpu_strategy",
579 srcs = ["tpu_strategy.py"],
643 ":tpu_strategy",
659 ":tpu_strategy",
893 ":tpu_strategy",
919 ":tpu_strategy",
1050 ":tpu_strategy",
1251 ":tpu_strategy",
1329 ":tpu_strategy",
1565 ":tpu_strategy",
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_encoding_distribution_test.py26 from tensorflow.python.distribute import tpu_strategy
43 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
Dcategory_crossing_distribution_test.py26 from tensorflow.python.distribute import tpu_strategy
43 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
/external/tensorflow/tensorflow/python/keras/distribute/
Dstrategy_combinations.py23 strategy_combinations.tpu_strategy,
56 strategy_combinations.tpu_strategy,
Dtpu_strategy_test_utils.py24 from tensorflow.python.distribute import tpu_strategy
49 return tpu_strategy.TPUStrategy(resolver)
Dkeras_rnn_model_correctness_test.py26 from tensorflow.python.distribute import tpu_strategy
126 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)):
Dkeras_correctness_test_base.py28 from tensorflow.python.distribute import tpu_strategy
108 strategy_combinations.tpu_strategy,
335 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
365 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)))
547 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
Dminimize_loss_test.py84 distribution=[strategy_combinations.tpu_strategy],
88 distribution=[strategy_combinations.tpu_strategy],
168 distribution=[strategy_combinations.tpu_strategy],
242 distribution=[strategy_combinations.tpu_strategy],
324 distribution=[strategy_combinations.tpu_strategy],
329 distribution=[strategy_combinations.tpu_strategy],
419 distribution=[strategy_combinations.tpu_strategy],
Dcheckpointing_test.py45 strategy_combinations.tpu_strategy,
105 strategy_combinations.tpu_strategy,
Dkeras_utils_test.py32 from tensorflow.python.distribute import tpu_strategy
106 if (isinstance(distribution, tpu_strategy.TPUStrategyV1) and
477 (tpu_strategy.TPUStrategy, tpu_strategy.TPUStrategyV1)) and
/external/tensorflow/tensorflow/python/distribute/experimental/
D__init__.py25 from tensorflow.python.distribute import tpu_strategy
DBUILD18 "//tensorflow/python/distribute:tpu_strategy",
/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/
Dmnist_train_test.py27 strategy_combinations.tpu_strategy,
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.distribute.combinations.pbtxt60 name: "tpu_strategy"

12