- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 376 for r2devices (0.15 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/device_context_test.cc
auto device_factory = DeviceFactory::GetFactory(device_type); 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]); tensorflow::AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_cluster_util.cc
std::function<WalkResult(Operation*, tf_device::ClusterOp, std::optional<std::string>)> callback) { mlir::TF::RuntimeDevices devices; if (failed(tensorflow::GetDevicesFromOp(module, &devices))) return failure(); const CallGraph call_graph(module); // symbol_table caches callees in the CallGraph. SymbolTableCollection symbol_table;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 4.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) -
pkg/volume/util/device_util_unsupported.go
return portalHostMap, nil } // FindDevicesForISCSILun unsupported returns nil func (handler *deviceHandler) FindDevicesForISCSILun(targetIqn string, lun int) ([]string, error) { devices := []string{} return devices, nil
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Aug 24 19:47:49 UTC 2021 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
return tensorflow::WrapOpInLaunch(builder, compile_op.getLoc(), compile_op, compilation_device); } // Assigns explicit devices to replicate op. An aliased device is created per // core, and all replica devices per core are grouped together. void AssignDevicesToReplicate( tf_device::ReplicateOp replicate, llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K 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) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/testdata/spmd.mlir
module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:localhost/replica:0/task:0/device:CPU:0", "/job:localhost/replica:0/task:0/device:TPU_SYSTEM:0", "/job:localhost/replica:0/task:0/device:TPU:0", "/job:localhost/replica:0/task:0/device:TPU:1", "/job:localhost/replica:0/task:0/device:TPU:2", "/job:localhost/replica:0/task:0/device:TPU:3", "/job:localhost/replica:0/task:0/device:TPU:4", "/job:localhost/replica:0/task:0/device:TPU:5", "/job:localhost/replica:0/task:0/device:TPU:6",...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 1.5K bytes - Viewed (0)