- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetDeviceOrdinalFromDeviceString (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/replicate_to_island.cc
<< " to be present in 'tf.device.replicate' op"; } llvm::StringRef tpu_device = tpu_replica.cast<ArrayAttr>()[replica_id].cast<StringAttr>().getValue(); return tensorflow::GetDeviceOrdinalFromDeviceString(op->getLoc(), tpu_device, &device_ordinal); } // Updates replica variant ops in a region based on replica `replica_id`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 24 21:01:40 UTC 2023 - 16.9K bytes - Viewed (0)