- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for ParseFullName (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
if (!string_attr) 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",
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
DeviceNameUtils::ParsedName parsed_name; DeviceNameUtils::ParseFullName(gpu0, &parsed_name); auto meta_0 = devices.GetGpuDeviceMetadata(parsed_name); ASSERT_TRUE(meta_0.has_value()); ASSERT_EQ(meta_0->getCcMajor(), 1); ASSERT_EQ(meta_0->getCcMinor(), 2); DeviceNameUtils::ParseFullName(gpu1, &parsed_name); auto meta_1 = devices.GetGpuDeviceMetadata(parsed_name);
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/tensorflow/transforms/rewrite_util.cc
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; } } // namespace
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/tf2xla/internal/passes/hoist_broadcast_read.cc
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 << "'"; } is_cpu_read = parsed_name.type == kCpuDeviceType; return success(); } }
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
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; auto attr_value = attr.second;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0)