- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for ParsedName (0.14 sec)
-
tensorflow/compiler/jit/device_util.cc
// TODO(sanjoy): Cache DeviceNameUtils::ParsedName inside device_info_cache. const auto is_multiple_devices = [&](const jit::DeviceId& d0, std::optional<jit::DeviceId>* d1) -> bool { const absl::string_view name0 = device_info_cache.GetNameFor(d0); const absl::string_view name1 = device_info_cache.GetNameFor(d1->value()); DeviceNameUtils::ParsedName parsed0, parsed1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
mlir::LogicalResult GetDevicesFromOp(mlir::Operation* op, mlir::ArrayAttr array_attr, mlir::TF::RuntimeDevices* devices) { DeviceNameUtils::ParsedName device; for (const auto& kv : llvm::enumerate(array_attr)) { const int idx = kv.index(); auto string_attr = mlir::dyn_cast<mlir::StringAttr>(kv.value()); if (!string_attr)
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/transforms/rewrite_util.cc
mlir::StringAttr device = op->getAttrOfType<mlir::StringAttr>(kDeviceAttr); if (!device || device.getValue().empty()) { return std::nullopt; } tensorflow::DeviceNameUtils::ParsedName parsed_name; if (!tensorflow::DeviceNameUtils::ParseFullName(device.str(), &parsed_name)) { return std::nullopt; } if (!parsed_name.has_type) { return std::nullopt; } return parsed_name.type; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
Status Sync() override { return errors::Unimplemented("FakeDevice::Sync()"); } static std::unique_ptr<Device> Make(const string& name, const string& desc = "") { DeviceNameUtils::ParsedName parsed_name; DeviceNameUtils::ParseFullName(name, &parsed_name); DeviceAttributes device_attributes; device_attributes.set_name(name); device_attributes.set_device_type(parsed_name.type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
if (auto attr = func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr)) { std::string device = attr.getValue().str(); tensorflow::DeviceNameUtils::ParsedName parsed_name; if (!tensorflow::DeviceNameUtils::ParseFullName(device, &parsed_name)) { return read->emitOpError() << "invalid device '" << device << "'"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
const std::string attrKey = "T"; for (const Node* node : graph.nodes()) { if (node->type_string() == nodeType) { auto device_name = node->assigned_device_name(); DeviceNameUtils::ParsedName device; if (DeviceNameUtils::ParseFullName(device_name, &device) && device.has_job && device.job == jobType) { for (const auto& attr : node->attrs()) { auto attr_key = attr.first;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)