- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 2,121 for Devices (0.14 sec)
-
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/mlir/tensorflow/ir/tf_ops_device_helper.h
class RuntimeDevices; // Returns true if at least one GPU device is available at runtime. bool CanUseGpuDevice(const RuntimeDevices &devices); // Returns true if all of the GPUs available at runtime support TensorCores // (NVIDIA compute capability >= 7.0). bool CanUseTensorCores(const RuntimeDevices &devices); // Returns true if operation does not have explicit device placement that would // prevent it from running on GPU device.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Nov 12 21:57:12 UTC 2021 - 1.4K bytes - Viewed (0) -
pkg/util/procfs/procfs_linux.go
if len(entries) == 3 && entries[1] == "devices" { return strings.TrimSpace(entries[2]), nil } } return "", fmt.Errorf("could not find devices cgroup location") } // GetFullContainerName gets the container name given the root process id of the container. // E.g. if the devices cgroup for the container is stored in /sys/fs/cgroup/devices/docker/nginx,
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Jan 16 09:22:35 UTC 2023 - 4.1K 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/mlir/tfrt/tests/runtime_lowering_tpu.mlir
// RUN: tf-tfrt-opt -tfrt-lower-cluster-to-runtime-ops-tpu -split-input-file -verify-diagnostics %s | FileCheck %s module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU_SYSTEM:0", "/job:worker/replica:0/task:0/device:TPU:0"]} { // CHECK-LABEL: @converts_cluster func.func @converts_cluster() { // CHECK: %0:2 = "tf_device.launch"() <{{.*}}> ({
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
SessionOptions options; std::vector<std::unique_ptr<Device>> devices; Status s = device_factory->CreateDevices( options, "/job:worker/replica:0/task:0", &devices); device_ = std::move(devices[0]); AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true); host_allocator_ = device_->GetAllocator(host_alloc_attr); AllocatorAttributes device_alloc_attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0) -
pkg/kubelet/cm/dra/claiminfo.go
func (info *ClaimInfo) cdiDevicesAsList() []kubecontainer.CDIDevice { var cdiDevices []kubecontainer.CDIDevice for _, devices := range info.CDIDevices { for _, device := range devices { cdiDevices = append(cdiDevices, kubecontainer.CDIDevice{Name: device}) } } return cdiDevices } // addPodReference adds a pod reference to the claim info.
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:30:31 UTC 2024 - 7.9K bytes - Viewed (0) -
pkg/util/procfs/procfs_fake.go
limitations under the License. */ package procfs type FakeProcFS struct{} // GetFullContainerName gets the container name given the root process id of the container. // E.g. if the devices cgroup for the container is stored in /sys/fs/cgroup/devices/docker/nginx, // return docker/nginx. Assumes that the process is part of exactly one cgroup hierarchy. func (fakePfs *FakeProcFS) GetFullContainerName(pid int) (string, error) { return "", nil
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Jan 16 09:22:35 UTC 2023 - 986 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/argument-sharding.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/defs.h
// The id of the compiled cluster. extern const char* const kXlaClusterIdAttr; // "_xla_compile_id" [[deprecated("XLA:CPU/GPU devices are deprecated")]] void RequestXlaDevicesCreation(); [[deprecated("XLA:CPU/GPU devices are deprecated")]] bool XlaDevicesCreationRequired(); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 07 01:03:32 UTC 2021 - 1.7K bytes - Viewed (0)