Home
last modified time | relevance | path

Searched refs:set_logical_devices_to_at_least (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dtest_util.py68 def set_logical_devices_to_at_least(device, num): function
95 set_logical_devices_to_at_least("GPU", 2)
97 set_logical_devices_to_at_least("CPU", 2)
Dtest_util_test.py81 test_util.set_logical_devices_to_at_least('CPU', 3)
Dstrategy_combinations.py329 test_util.set_logical_devices_to_at_least("CPU", num_virtual_cpus)
/external/tensorflow/tensorflow/python/ops/
Dnn_loss_scaling_utilities_test.py39 test_util.set_logical_devices_to_at_least("CPU", 3)
/external/tensorflow/tensorflow/python/kernel_tests/
Dcollective_ops_test.py1023 test_util.set_logical_devices_to_at_least('CPU', 4)
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dautocast_variable_test.py69 test_util.set_logical_devices_to_at_least('CPU', 3)