- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 376 for r2devices (0.11 sec)
-
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) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.h
==============================================================================*/ // This file defines the tf_device dialect: it contains operations that model // TensorFlow's actions to launch computations on accelerator devices. #ifndef TENSORFLOW_COMPILER_MLIR_TENSORFLOW_IR_TF_DEVICE_H_ #define TENSORFLOW_COMPILER_MLIR_TENSORFLOW_IR_TF_DEVICE_H_ #include "mlir/Dialect/Func/IR/FuncOps.h" // from @llvm-project
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 14:25:57 UTC 2022 - 2.2K bytes - Viewed (0) -
pkg/kubelet/cm/dra/claiminfo_test.go
expectedResult []kubecontainer.CDIDevice }{ { description: "empty CDI devices", claimInfo: &ClaimInfo{ ClaimInfoState: state.ClaimInfoState{ CDIDevices: map[string][]string{}, }, }, }, { description: "nil CDI devices", claimInfo: &ClaimInfo{}, }, { description: "valid CDI devices", claimInfo: &ClaimInfo{ ClaimInfoState: state.ClaimInfoState{
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri May 03 13:30:31 UTC 2024 - 21K bytes - Viewed (0) -
pkg/volume/util/volumepathhandler/volume_path_handler_linux.go
if err != nil { return "", fmt.Errorf("failed to evaluate path %s: %s", path, err) } devices, err := filepath.Glob("/sys/block/loop*") if err != nil { return "", fmt.Errorf("failed to list loop devices in sysfs: %s", err) } for _, device := range devices { backingFile := fmt.Sprintf("%s/loop/backing_file", device) // The contents of this file is the absolute path of "path".
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Tue Jun 20 14:49:03 UTC 2023 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/jit/flags.h
// 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. // This can only be modified programmatically. absl::flat_hash_set<std::string> xla_launch_allowed_devices_; // Devices for which using Device API (PjRt) is allowed in the // XlaCompileOnDemand op. This can only be modified programmatically.
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/jit/xla_kernel_creator_test.cc
SessionOptions options; auto* device_count = options.config.mutable_device_count(); device_count->insert({"CPU", 1}); std::vector<std::unique_ptr<Device>> devices; TF_CHECK_OK(DeviceFactory::AddDevices( options, "/job:localhost/replica:0/task:0", &devices)); FunctionDefLibrary proto; for (const auto& fdef : flib) { *(proto.add_function()) = fdef; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.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)