- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 847 for device_1 (0.19 sec)
-
tensorflow/compiler/jit/xla_platform_info.h
// configuring the persistor used in the DeviceCompiler. Please note that // non-XLA devices aren't supported yet. This is because: // 1. PjRtClient doesn't support data transfer for non-XLA devices yet // 2. Fetching the PjRtClient for non-XLA devices is also not supported yet Status GetOrCreatePjRtDeviceCompilerAndProfiler( const OpKernelContext& ctx, const XlaPlatformInfo& platform_info,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/test_util.cc
} Device* DeviceSetup::GetDevice(const string& device_name) { if (device_mgr_ == nullptr) { return nullptr; } string full_device_name = absl::StrCat( "/job:localhost/replica:0/task:0/device:", device_name, ":0"); Device* device; TF_CHECK_OK(device_mgr_->LookupDevice(full_device_name, &device)); return device; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/build_xla_ops_pass.cc
VLOG(3) << ndef.DebugString(); if (!ndef.device().empty()) { TF_ASSIGN_OR_RETURN(jit::DeviceId device_id, device_info_cache->GetIdFor(ndef.device())); device_set.Insert(device_id); } } if (!n->assigned_device_name().empty()) { // TODO(sanjoy): We need this because EncapsulateSubgraphsPass drops device
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replica_id_to_device_ordinal.cc
return signalPassFailure(); } // Get the number of devices per host. int device_num = 0; mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp( getOperation()->getParentOfType<ModuleOp>(), &devices))) return signalPassFailure(); for (const auto& device_name : devices.device_names()) { if (device_name.has_type && device_name.type == "TPU") ++device_num; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4K bytes - Viewed (0) -
tensorflow/compiler/jit/partially_decluster_pass_test.cc
// This is needed to register the XLA_GPU device. std::vector<std::unique_ptr<Device>> devices; TF_ASSERT_OK(DeviceFactory::AddDevices( SessionOptions(), "/job:localhost/replica:0/task:0", &devices)); // Scope::ToGraph loses the assigned device name since it goes through // GraphDef/NodeDef which does not have a field for the assigned device name. Node* n = FindNodeByName(*graph, "shape");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 10 12:32:39 UTC 2022 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
value_to_device.insert({result, device}); } } // Checks if an operation can have TPU devices propagated through. bool IsSupportedOpToSetDevice(Operation& op) { return IsSupportedExecutorOp(op) || isa<TF::IdentityOp, TF::IdentityNOp, TF::ShapeOp>(op); } // Finds nonconflicting TPU device for an operation from its operands. If an // operand has no device or a non TPU device, or if there are conflicting
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
pkg/kubelet/volumemanager/cache/actual_state_of_world_test.go
VolumeSource: v1.VolumeSource{ GCEPersistentDisk: &v1.GCEPersistentDiskVolumeSource{ PDName: "fake-device1", }, }, }, }, }, } volumeSpec := &volume.Spec{Volume: &pod.Spec.Volumes[0]} devicePath := "fake/device/path" deviceMountPath := "fake/device/mount/path" generatedVolumeName, err := util.GetUniqueVolumeNameFromSpec(plugin, volumeSpec) if err != nil {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 46.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
// product of possible devices and inference datatypes. For example, we might // raise a chunk of sequential operations from a block all having attributes // `{ tac.device = "GPU", tac.inference_type = "FLOAT"}` to a function // with the matching attributes. Assumed is that device type "CPU" // is the only device that is allowed to call other devices. I.e. ancestors of a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_ops.mlir
tf_device.replicate() {n = 2 : i32, devices = {TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"], TPU_REPLICATED_CORE_1 = ["/DEVICE:2", "/DEVICE:3"]}} { tf_device.return } func.return // CHECK: tf_device.replicate // CHECK-SAME: devices = {TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"], TPU_REPLICATED_CORE_1 = ["/DEVICE:2", "/DEVICE:3"]} // CHECK-SAME: n = 2 // CHECK-NEXT: tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
Device* device, Tensor* output_tensor, StatusCallback done) const { done(errors::Unimplemented("XLA->XLA same-device copies not implemented.")); } void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0)