Home
last modified time | relevance | path

Searched refs:VariableDeviceChooser (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/
Dhybrid_layer.py36 device_assigner or framework_variables.VariableDeviceChooser())
Dhybrid_model.py49 device_assigner or framework_variables.VariableDeviceChooser())
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py485 device_fn = variables_lib2.VariableDeviceChooser(num_tasks=2)
514 device_fn = variables_lib2.VariableDeviceChooser(replica=3, num_tasks=2)
543 device_fn = variables_lib2.VariableDeviceChooser(device_type='GPU')
658 device_fn = variables_lib2.VariableDeviceChooser()
Dvariables.py757 class VariableDeviceChooser(object): class
/external/tensorflow/tensorflow/contrib/tensor_forest/python/
Dtensor_forest.py381 device_assigner or framework_variables.VariableDeviceChooser())