- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for device_num (0.44 sec)
-
tensorflow/c/kernels_test.cc
static std::unique_ptr<OpKernel> GetFakeKernel(const char* device_name, const char* op_name, const char* node_name, Status* status) { NodeDef def; def.set_op(op_name); def.set_name(node_name); def.set_device(device_name); def.add_input("input1"); def.add_input("input2");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 50.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
const char* device_name = TFE_OpGetDevice(matmul, status); ASSERT_TRUE(strstr(device_name, "GPU:0") != nullptr); TFE_OpSetDevice(matmul, "CPU:0", status); ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status); device_name = TFE_OpGetDevice(matmul, status); ASSERT_TRUE(strstr(device_name, "CPU:0") != nullptr); } TFE_DeleteOp(matmul);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
for (int i = 0; i < num_devices; ++i) { const char* device_name = TF_DeviceListName(device_list, i, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); const char* device_type = TF_DeviceListType(device_list, i, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); LOG(INFO) << "Device " << i << " has name " << device_name << ", type " << device_type; if (string(device_type) == DEVICE_GPU) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 96.9K bytes - Viewed (0)