- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GetDeviceAliasForHostOfLogicalCore (0.4 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
std::string& host_device) { mlir::tf_device::ReplicateOp replicate = cluster->getParentOfType<mlir::tf_device::ReplicateOp>(); if (replicate) { host_device = GetDeviceAliasForHostOfLogicalCore(0); return mlir::success(); } tensorflow::TPUDevicesAndHosts devices_and_hosts; if (failed(GetTPUDevicesAndHostsNotReplicated(devices, cluster,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
assigned_id->setAttr(kICIWeightDistributionMlirBridgeMarker, before_cluster_builder.getBoolAttr(true)); std::string device = tensorflow::GetDeviceAliasForHostOfLogicalCore(0); LaunchOp launch = tensorflow::WrapOpInLaunch( &before_cluster_builder, val_bcast.getLoc(), assigned_id, device); Value all_reduce =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)