- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetGpuDeviceMetadata (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_structs.cc
device_names_.push_back(device); gpu_metadata_.insert({DeviceNameUtils::ParsedNameToString(device), metadata}); } std::optional<GpuDeviceMetadata> RuntimeDevices::GetGpuDeviceMetadata( const ParsedName& device) const { auto it = gpu_metadata_.find(DeviceNameUtils::ParsedNameToString(device)); if (it != gpu_metadata_.end()) { return it->second; } else { return std::nullopt;
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_test.cc
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); ASSERT_FALSE(meta_1.has_value()); }
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/ir/tf_structs.h
size_t NumDevices() const { return device_names_.size(); } // Returns GPU device metadata if it is available, otherwise returns None. std::optional<GpuDeviceMetadata> GetGpuDeviceMetadata( const ParsedName& device) const; private: llvm::SmallVector<ParsedName, 8> device_names_; // TODO(ezhulenev): Add DenseMapInfo<ParsedName> specialization to be able to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 2.6K bytes - Viewed (0)