- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 1,341 for Devices (0.14 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
auto& side_effect_analysis = getAnalysis<mlir::TF::SideEffectAnalysis>(); // Get runtime devices information from the closest parent module. auto module = getOperation(); mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp(module, &devices))) return signalPassFailure(); OpBuilder builder(&getContext()); llvm::SmallVector<mlir::tf_device::ClusterOp, 4> clusters;
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/jit/flags.h
// `enabled_for_gpu_` below. bool enabled_for_all_; // If true, enable Device API (PjRt) for TF GPU device. This is a helper // flag so that individual tests can turn on PjRt for GPU specifically. // Once the rollout to GPU is complete, this flag can be deprecated. bool enabled_for_gpu_; private: // Devices for which using Device API (PjRt) is allowed in the XlaLaunch op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 14.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
pkg/kubelet/cm/dra/manager.go
annotations = append(annotations, claimAnnotations...) devices := claimInfo.cdiDevicesAsList() klog.V(3).InfoS("Add CDI devices", "claim", *claimName, "CDI devices", devices) cdiDevices = append(cdiDevices, devices...) return nil }) if err != nil { return nil, fmt.Errorf("locked cache operation: %w", err) } } }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:23:29 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
static const char* const module_str = R"( module attributes {tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:GPU:0"]} { func.func @convert_cluster_func(%arg0: tensor<i32>) -> () { %2 = "tf_device.parallel_execute"() ({ %3 = "tf_device.cluster_func"(%arg0) {device = "/job:localhost/replica:0/task:0/device:GPU:0", func = @func} : (tensor<i32>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.cc
"Switch a device into 'on-demand' mode, where instead of " "autoclustering ops are compiled one by one just-in-time."), Flag("tf_xla_enable_xla_devices", &device_flags->tf_xla_enable_xla_devices, "Generate XLA_* devices, where placing a computation on such a " "device" "forces compilation by XLA. Deprecated."),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 24.5K 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/tfrt/tests/tf_to_corert/tf_to_corert_pipeline_refvar.mlir
// CHECK-NEXT: [[o_chain:%.*]], [[o:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) key(0) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.VarHandleOp"() // CHECK-NEXT: [[o_chain_0:%.*]], [[o1:%.*]] = tfrt_fallback_async.executeop.seq([[in_chain]]) key(1) cost({{.*}}) device("/job:localhost/replica:0/task:0/device:CPU:0") "tf.ReadVariableOp"([[o]]) {dtype = f32} : 1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
// For EnqueueTPUEmbedding ops the device ordinal corresponds to the resource // instance. We also take the `device` attribute into account in order to avoid // dependencies between ops with the same ordinal on different devices. // Helper function to get an absolute device string, combining device and // ordinal attribute values.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
pkg/kubelet/cm/dra/state/state_checkpoint.go
PodUIDs sets.Set[string] // ResourceHandles is a list of opaque resource data for processing by a specific kubelet plugin ResourceHandles []resourcev1alpha2.ResourceHandle // CDIDevices is a map of DriverName --> CDI devices returned by the // GRPC API call NodePrepareResource CDIDevices map[string][]string } // ClaimInfoStateWithoutResourceHandles is an old implementation of the ClaimInfoState // TODO: remove in Beta
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:23:27 UTC 2024 - 4.3K bytes - Viewed (0)