Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for kDevicesAttr (0.09 sec)

  1. 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)
  2. 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)
Back to top