Searched refs:get_device_name (Results 1 – 10 of 10) sorted by relevance
209 auto get_device_name = [&placer](const string& device) -> string { in TEST() local219 get_device_name("/replica:0/cpu:0")); in TEST()221 get_device_name("/replica:2/cpu:0")); in TEST()223 get_device_name("/replica:7/cpu:0")); in TEST()225 get_device_name("/replica:3/gpu:0")); in TEST()227 get_device_name("/replica:5/gpu:3")); in TEST()229 get_device_name("/replica:4/gpu:7")); in TEST()240 get_device_name("/job:worker/replica:0/cpu:0")); in TEST()242 get_device_name("/job:worker/replica:7/gpu:3")); in TEST()244 get_device_name("/job:ps/replica:0/cpu:0")); in TEST()[all …]
49 std::string get_device_name(cl_device_id device);
119 std::string get_device_name(cl_device_id device) in get_device_name() function
344 clDeviceInfoStream << "CL_DEVICE_NAME=\"" << get_device_name(device).c_str() in get_cl_device_info_str()
12 name: "get_device_name"
1079 static const char *get_device_name(const char *fs_type) in get_device_name() function1107 SAFE_MOUNT(get_device_name(tdev.fs_type), tst_test->mntpoint, in prepare_device()
22 from tensorflow.python.eager.context import get_device_name1757 current_device = get_device_name()
782 self.assertEmpty(context.get_device_name())
2336 def get_device_name(): function
509 py::object get_device_name() const { in get_device_name() function in __anon7071e87e0411::EagerContextThreadLocalDataWrapper1675 &EagerContextThreadLocalDataWrapper::get_device_name, in PYBIND11_MODULE()