Searched refs:VariableDeviceChooser (Results 1 – 5 of 5) sorted by relevance
36 device_assigner or framework_variables.VariableDeviceChooser())
49 device_assigner or framework_variables.VariableDeviceChooser())
485 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()
757 class VariableDeviceChooser(object): class
381 device_assigner or framework_variables.VariableDeviceChooser())