- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TPUDeviceAssignment (0.65 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// `device_assignment` are provided. struct TPUDeviceAssignment { TPUDeviceAssignment(llvm::StringRef compilation_device, TPUDevicesAndHosts&& tpu_devices) : compilation_device(compilation_device), tpu_devices(std::move(tpu_devices)) {} TPUDeviceAssignment(llvm::StringRef compilation_device, TPUDevicesAndHosts&& tpu_devices,
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 (failed(GetDeviceAssignmentCoordinates(cluster, device_coordinates))) { return mlir::failure(); } // Determine compilation and execution devices. if (absl::StatusOr<TPUDeviceAssignment> tpu_device_assignment = tensorflow::GetTPUCompilationAndExecutionDevices( devices.device_names(), /*num_replicas=*/1, GetNumCoresPerReplica(cluster), topology, device_coordinates);
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
return llvm::formatv("requires attribute '{0}'", attribute).str(); } LogicalResult GetTpuDeviceAssignment( ClusterOp cluster, ReplicateOp replicate, mlir::ModuleOp module, absl::StatusOr<TPUDeviceAssignment>& status_or_tpu_device_assignment) { mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp(module, &devices))) return failure(); uint32_t num_replicas = replicate.getN();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)