Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/distribute/python/
Done_device_strategy.py21 from tensorflow.python.distribute import one_device_strategy
23 OneDeviceStrategy = one_device_strategy.OneDeviceStrategy
Dmonitor_test.py28 from tensorflow.python.distribute import one_device_strategy
64 distribution = one_device_strategy.OneDeviceStrategy(
74 distribution = one_device_strategy.OneDeviceStrategy(
Dcombinations.py54 from tensorflow.python.distribute import one_device_strategy as one_device_lib
352 one_device_strategy = NamedDistribution( variable
448 one_device_strategy,
461 one_device_strategy,
Dcheckpoint_utils_test.py67 combinations.one_device_strategy,
101 combinations.default_strategy, combinations.one_device_strategy,
DBUILD127 name = "one_device_strategy",
128 srcs = ["one_device_strategy.py"],
131 "//tensorflow/python/distribute:one_device_strategy",
191 ":one_device_strategy",
515 "//tensorflow/python/distribute:one_device_strategy",
Dwarm_starting_util_test.py45 combinations.one_device_strategy,
Done_device_strategy_test.py30 combinations.one_device_strategy,
Destimator_integration_test.py63 combinations.one_device_strategy,
Dmoving_averages_test.py34 combinations.one_device_strategy,
Dkeras_utils_test.py324 distribution=[combinations.one_device_strategy], mode=['graph']))
350 distribution=[combinations.one_device_strategy], mode=['eager']))
Dkeras_correctness_test_base.py46 combinations.one_device_strategy,
Dmetrics_v1_test.py78 combinations.one_device_strategy,
Dminimize_loss_test.py276 combinations.one_device_strategy,
Dkeras_test.py265 combinations.one_device_strategy,
294 combinations.one_device_strategy,
Dcross_device_ops_test.py220 combinations.one_device_strategy,
Dkeras_backward_compat_test.py294 combinations.one_device_strategy,
/external/tensorflow/tensorflow/python/keras/mixed_precision/experimental/
Dloss_scale_optimizer_test.py24 from tensorflow.python.distribute import one_device_strategy
36 return one_device_strategy.OneDeviceStrategy('cpu:0')
DBUILD110 "//tensorflow/python/distribute:one_device_strategy",
132 "//tensorflow/python/distribute:one_device_strategy",
Dkeras_test.py25 from tensorflow.python.distribute import one_device_strategy
124 return one_device_strategy.OneDeviceStrategy('cpu:0')
/external/tensorflow/tensorflow/python/distribute/
DBUILD118 ":one_device_strategy",
272 name = "one_device_strategy",
273 srcs = ["one_device_strategy.py"],
417 ":one_device_strategy",
D__init__.py27 from tensorflow.python.distribute import one_device_strategy
/external/tensorflow/tensorflow/contrib/distribute/
D__init__.py30 from tensorflow.contrib.distribute.python.one_device_strategy import OneDeviceStrategy
DBUILD31 "//tensorflow/contrib/distribute/python:one_device_strategy",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-one-device-strategy.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.one_device_strategy.OneDeviceStrategy\'>"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-one-device-strategy.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.one_device_strategy.OneDeviceStrategy\'>"