- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TPU_REPLICATED_CORE (0.53 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
tensorflow::DeviceNameUtils::ParsedName name; if (tensorflow::DeviceNameUtils::ParseFullName(device.str(), &name)) { if (name.type == "TPU_REPLICATED_CORE") { // TODO(hanxiongwang): Add check for out of bound of name.id return llvm::StringRef(logical_device_vec[name.id]); } } } } return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
// topology. constexpr int kTPUTopologyRank = 4; constexpr char kDeviceTPUSystem[] = "TPU_SYSTEM"; constexpr char kDeviceTPU[] = "TPU"; constexpr char kTPUReplicatedCore[] = "TPU_REPLICATED_CORE"; constexpr char kTPUReplicatedHost[] = "TPU_REPLICATED_HOST"; constexpr char kBadIntArrayElementMsg[] = "bad '{0}' attribute at index {1}, not an int"; using ParsedDevice = DeviceNameUtils::ParsedName;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0)