- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for addService (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_structs.cc
==============================================================================*/ #include "tensorflow/compiler/mlir/tensorflow/ir/tf_structs.h" #include <optional> namespace mlir { namespace TF { void RuntimeDevices::AddDevice(const ParsedName& device) { device_names_.push_back(device); } void RuntimeDevices::AddGpuDevice(const ParsedName& device, const GpuDeviceMetadata& metadata) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 1.4K bytes - Viewed (0) -
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/ir/tf_structs.h
using DeviceNameUtils = ::tensorflow::DeviceNameUtils; using ParsedName = ::tensorflow::DeviceNameUtils::ParsedName; public: // Adds a device with and empty metadata. Device can be of any type. void AddDevice(const ParsedName& device); // Adds a GPU device with GPU specific metadata. void AddGpuDevice(const ParsedName& device, const GpuDeviceMetadata& metadata);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 2.6K 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)