- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for num_devices (0.28 sec)
-
tensorflow/compiler/jit/device_util_test.cc
CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); } void SimpleRoundTripTestForDeviceSet(int num_devices) { jit::DeviceSet device_set; jit::DeviceInfoCache device_info_cache; std::vector<string> expected_devices, actual_devices; for (int i = 0; i < num_devices; i++) { string device_name = absl::StrCat("/job:localhost/replica:0/task:0/device:XPU:", i);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island_legacy.mlir
// CHECK: %{{.*}} = tf_executor.island(%[[CT_0]], %[[CT_1]]) // Tests devices are not remapped if no devices were defined in replicate. // CHECK-LABEL: func @no_devices func.func @no_devices() { tf_executor.graph { %0 = tf_executor.island { tf_device.replicate {n = 2 : i32} { "tf_device.launch"() ({ "tf.opA"() : () -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
(*module_ref)->setAttr("tf.devices", device_dict); mlir::TF::RuntimeDevices devices; EXPECT_TRUE(mlir::succeeded(GetDevicesFromOp(*module_ref, &devices))); ASSERT_EQ(devices.NumDevices(), 1); ASSERT_EQ(devices.device_names().size(), 1); ASSERT_EQ(DeviceNameUtils::ParsedNameToString(devices.device_names()[0]), "/job:worker/replica:0/task:0/device:CPU:0"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0)