- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 268 for r2devices (0.14 sec)
-
pkg/kubelet/cm/devicemanager/plugin/v1beta1/stub.go
klog.InfoS("ListAndWatch") s.Send(&pluginapi.ListAndWatchResponse{Devices: m.devs}) for { select { case <-m.stop: return nil case updated := <-m.update: s.Send(&pluginapi.ListAndWatchResponse{Devices: updated}) } } } // Update allows the device plugin to send new devices through ListAndWatch func (m *Stub) Update(devs []*pluginapi.Device) { m.update <- devs }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Oct 17 11:19:10 UTC 2023 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
Builder* builder, OperationState* state, int n, std::optional<DictionaryAttr> devices, llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, ValueRange packed_inputs, TypeRange replica_output_types) { DCHECK_GE(n, 2); state->addAttribute("n", builder->getI32IntegerAttr(n)); if (devices.has_value()) state->addAttribute("devices", devices.value()); Region* region = state->addRegion();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_device_ops.mlir
// ----- // CHECK-LABEL: func @replicate_with_devices func.func @replicate_with_devices() { tf_device.replicate() {n = 2 : i32, devices = {TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"]}} { tf_device.return } func.return // CHECK: tf_device.replicate // CHECK-SAME: devices = {TPU_REPLICATED_CORE_0 = ["/DEVICE:0", "/DEVICE:1"]} // 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/tensorflow/tests/tpu-variable-runtime-reformatting.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 25.4K bytes - Viewed (0) -
pkg/kubelet/cm/dra/claiminfo.go
prepared: false, } for pluginName, devices := range info.CDIDevices { annotations, _ := cdi.GenerateAnnotations(info.ClaimUID, info.DriverName, devices) info.annotations[pluginName] = append(info.annotations[pluginName], annotations...) } return info } // setCDIDevices adds a set of CDI devices 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) -
tensorflow/c/eager/c_api_test_util.cc
TF_NewStatus(), TF_DeleteStatus); TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); CHECK_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); const int num_devices = TF_DeviceListCount(devices); for (int i = 0; i < num_devices; ++i) { const string dev_type(TF_DeviceListType(devices, i, status.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/saved_variable_loading_test.cc
SessionOptions options; options.config.mutable_device_count()->insert({"CPU", 3}); std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices)); device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); ctx_ = testing::CreateTestingEagerContext(device_mgr_.get()); } EagerContext* context() { return ctx_.get(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 09:34:33 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_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]); AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true); host_allocator_ = device_->GetAllocator(host_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) -
tensorflow/c/eager/c_api_experimental.h
size_t proto_len, TF_Status* status); // TODO(b/166642410): It would be nice, for custom devices and for other users, // to have a non-string representation of devices (TF_Device) extracted from // tensors/ops/etc. and usable in APIs like OpSetDevice/ResetOp/etc. #define TFE_CUSTOM_DEVICE_VERSION 4
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 39.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
} return resource_users; } void ColocateCompositeResourceOpsInReplicate( tf_device::ReplicateOp replicate_op, OpBuilder* builder) { auto devices = replicate_op.getDevices(); if (!devices) return; if (!devices.value().get(tensorflow::GetDeviceAliasForLogicalCore(0))) return; const auto composite_resource_users = GetResourceOpsUsingCompositeArgsInReplicate(replicate_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0)