- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for kDevicesAttr (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
"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())); } } return mlir::success(); }
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/host_runtime/tpu_rewrite_pass.cc
#include "tensorflow/core/util/device_name_utils.h" namespace mlir { namespace TFTPU { constexpr char kStepMarkerLocationAttr[] = "step_marker_location"; constexpr char kDevicesAttr[] = "devices"; constexpr char kVersionsAttr[] = "tf.versions"; constexpr char kUseXlaSpmdAttr[] = "use_spmd_for_xla_partitioning"; constexpr char kBadStringArrayElementMsg[] =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)