- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 342 for Devices (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
task_and_device = {task, device}; } } return topology; } // Determine execution devices when topology and device assignment are defined. // With a topology device coordinate to task and device mapping, device // assignment device coordinates can then be mapped to task and device for TPU // devices. The device assignment array is also validated. //
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
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(); auto num_cores_per_replica_attr = cluster->getAttrOfType<mlir::IntegerAttr>(
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/tests/xla_broadcast.mlir
// RUN: tf-opt %s -split-input-file -tf-xla-broadcast | FileCheck %s
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/sink_variable_as_named_array.mlir
// RUN: tf-tfrt-opt -split-input-file -tf-device-decompose-resource-ops -sink-variable-as-named-array %s | FileCheck %s // ----- // Basic test: all variables tensors are for devices and sinked as named ifrt arrays // // // CHECK-LABEL: func.func @serving_default(%arg0: tensor<1x3xf32>) -> tensor<1x1xf32> { // CHECK-NEXT: [[HANDLE2:%.*]] = "tf.VarHandleOp" // CHECK-NEXT: [[KEY:%.*]], [[FUTURE:%.*]] = "tf.IfrtLoadVariable"([[HANDLE2]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 15:33:17 UTC 2024 - 5.3K bytes - Viewed (0) -
cluster/gce/gci/configure-helper.sh
return fi local device="${devices[0]}" if [ "${#devices[@]}" -ne 1 ]; then seen_arrays=(/dev/md/*) device=${seen_arrays[0]} echo "Setting RAID array with local SSDs on device ${device}" if [ ! -e "$device" ]; then device="/dev/md/0" echo "y" | mdadm --create "${device}" --level=0 --raid-devices=${#devices[@]} "${devices[@]}" fi fi
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Jun 10 22:07:47 UTC 2024 - 141.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/BUILD
":__subpackages__", "//tensorflow/python:__subpackages__", ], deps = [ "@llvm-project//mlir:AffineDialect", "@llvm-project//mlir:QuantOps", # Link jit lib to link JIT devices required to run # xla-legalize-tf-with-tf2xla pass. "//tensorflow/compiler/jit", "//tensorflow/compiler/mlir/lite:tensorflow_lite", "//tensorflow/compiler/mlir/lite:tensorflow_lite_legalize_tf",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
std::unique_ptr<OperationPass<mlir::ModuleOp>> CreateCrossHostTransferPass(); // Creates a pass that adds the device attribute to every tf.Const op based on // the device attribute of the operations that read its result. If the result of // a tf.Const op is read by operations placed on multiple devices, then the pass // will replicate the tf.Const op once for each device. std::unique_ptr<OperationPass<ModuleOp>> CreateConstantOpDeviceAssignmentPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
pkg/kubelet/container/runtime.go
Envs []EnvVar // The mounts for the containers. Mounts []Mount // The host devices mapped into the containers. Devices []DeviceInfo // The CDI devices for the container CDIDevices []CDIDevice // The annotations for the container // These annotations are generated by other components (i.e., // not users). Currently, only device plugins populate the annotations. Annotations []Annotation
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Jun 07 00:05:23 UTC 2024 - 27.8K bytes - Viewed (0) -
pkg/kubelet/kubelet_pods.go
return m, nil } services, err := kl.serviceLister.List(labels.Everything()) if err != nil { return m, fmt.Errorf("failed to list services when setting up env vars") } // project the services in namespace ns onto the master services for i := range services { service := services[i] // ignore services where ClusterIP is "None" or empty if !v1helper.IsServiceIPSet(service) {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Jun 14 16:09:17 UTC 2024 - 101.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
} def SimpleTFDeviceAssignmentPass : Pass<"tf-simple-device-assignment", "mlir::func::FuncOp"> { let summary = "Simple device assignment in TF dialect."; let constructor = "TF::CreateSimpleTFDeviceAssignmentPass()"; let options = [ Option<"default_device_", "default-device", "std::string", /*default=*/"\"cpu\"", "The default device to assign."> ]; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)