- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for getAllowSoftPlacement (0.28 sec)
-
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)