- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for AddDevice (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
return op->emitOpError(llvm::formatv( "bad '{0}' attribute at index {1}, not a string", kDevicesAttr, idx)); if (DeviceNameUtils::ParseFullName(string_attr.getValue().str(), &device)) { devices->AddDevice(device); } else { return op->emitOpError( llvm::formatv("bad '{0}' attribute, '{1}', not a valid device", kDevicesAttr, string_attr.getValue())); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
devices.push_back(FakeDevice::Make(gpu0, "compute capability: 7.0")); devices.push_back(FakeDevice::Make(gpu1)); DeviceSet device_set; for (auto& device : devices) device_set.AddDevice(device.get()); AddDevicesToOp(*module_ref, &device_set); auto devices_attr = (*module_ref)->getAttrOfType<mlir::DictionaryAttr>("tf.devices"); ASSERT_NE(devices_attr, nullptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0)