Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for tf_mlir_enable_multiple_local_cpu_devices (0.35 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc 

        ParsedDevice tpu_device, ParsedDevices devices) {
  tpu_device.type = DEVICE_CPU;
  bool enable_multiple_local_cpu_devices =
      tensorflow::GetMlirCommonFlags()
          ->tf_mlir_enable_multiple_local_cpu_devices;
  if (!enable_multiple_local_cpu_devices) {
    tpu_device.id = 0;
  }
  if (FindMatchingDevices(devices, tpu_device).empty()) {
    return absl::InvalidArgumentError(absl::StrCat(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 10 20:10:40 UTC 2024
    - 32.8K bytes
    - Viewed (0)
Back to top