- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for GetHostDeviceOutsideComputation (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); mlir::TF::RuntimeDevices devices; std::string host_device; EXPECT_TRUE(mlir::failed( GetHostDeviceOutsideComputation(devices, cluster, &host_device))); } TEST(TPURewriteDeviceUtilTest, TestGetHostFailDeviceMissingAttributes) { 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) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_cluster_util.cc
auto result = module.walk([&](tf_device::ClusterOp tpu_cluster) { std::optional<std::string> host_device; if (pass_host_device) { std::string host_device_value; if (failed(tensorflow::GetHostDeviceOutsideComputation( devices, tpu_cluster, &host_device_value))) return WalkResult::interrupt(); host_device = host_device_value; } cluster_to_host[tpu_cluster.getOperation()] = host_device;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
FindOutsideCompiledOpsAtHead(side_effect_analysis, cluster); if (head_outside_compiled_ops.empty()) return mlir::success(); if (failed(tensorflow::GetHostDeviceOutsideComputation(devices, cluster, host_device))) return mlir::failure(); CreateHeadComputation(builder, cluster, head_outside_compiled_ops,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// returns the host device for the head and tail computations. For TPU device, // if the computation is replicated, GetDeviceAliasForHostOfLogicalCore(0) is // returned instead. mlir::LogicalResult GetHostDeviceOutsideComputation( mlir::TF::RuntimeDevices devices, mlir::tf_device::ClusterOp cluster, std::string* host_device); // Checks if a device string is a TPU device. bool IsTPUDevice(llvm::StringRef device);
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
} } if (host_device->empty()) { LOG(ERROR) << "Did not find any CPU:0 host devices"; return mlir::failure(); } return mlir::success(); } mlir::LogicalResult GetHostDeviceOutsideComputation( mlir::TF::RuntimeDevices devices, mlir::tf_device::ClusterOp cluster, std::string* host_device) { if (HasTPUDevice(devices) || cluster->getParentOfType<mlir::tf_device::ReplicateOp>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0)