Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getAllowSoftPlacement (0.28 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tpu_validate_inputs.cc

      auto cluster = op->getAttrOfType<StringAttr>(TF::kTpuReplicateAttr).str();
      if (metadata_map.find(cluster) == metadata_map.end()) return false;
      auto metadata = metadata_map[cluster];
      if (!metadata.getAllowSoftPlacement() &&
          !op->hasAttr(TF::kXlaOutsideCompilationAttr))
        return true;
      std::string device = "";
      if (op->hasAttr(TF::kDeviceAttr))
        device = op->getAttrOfType<StringAttr>(TF::kDeviceAttr).str();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 06:51:01 UTC 2024
    - 21.5K bytes
    - Viewed (0)
Back to top