- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetDeviceAliasForHostOfLogicalCore (0.56 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
std::string GetDeviceAliasForLogicalCore(int core_index); // Virtual device name of the host that is associated with the passed logical // core. The logical core is the index of a core within a replica. std::string GetDeviceAliasForHostOfLogicalCore(int core_index); // Returns true if cluster contains model parallelism based on // `num_cores_per_replica_attribute`. Otherwise returns false. bool HasModelParallelism(mlir::tf_device::ClusterOp cluster);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
// For data parallelism, also add replicated host devices, as these are // necessary for outside compilation. device_attrs.push_back(builder->getNamedAttr( tensorflow::GetDeviceAliasForHostOfLogicalCore(core), builder->getStrArrayAttr(hosts_by_core))); } replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
mlir::TF::RuntimeDevices runtime_devices; std::string host_device; EXPECT_TRUE(mlir::succeeded( GetHostDeviceOutsideComputation(runtime_devices, cluster, &host_device))); EXPECT_EQ(host_device, GetDeviceAliasForHostOfLogicalCore(0)); } TEST(TPURewriteDeviceUtilTest, TestGetHostDeviceNotReplicated) { mlir::MLIRContext context; context.loadDialect<mlir::tf_device::TensorFlowDeviceDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0)