- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 1,050 for Devices (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
mlir::Builder builder(ctx); // Collect devices with attached metadata. llvm::SmallVector<mlir::NamedAttribute, 8> devices; devices.reserve(device_set->devices().size()); // For device that do not have any metadata, or if we failed to parse metadata // from the DeviceSet, we add a unit attribute to the `tf.devices` attribute. for (Device* device : device_set->devices()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util_test.cc
llvm::SmallVector<std::unique_ptr<Device>, 2> devices; devices.push_back(FakeDevice::Make(cpu0)); devices.push_back(FakeDevice::Make(gpu0, "compute capability: 7.0")); devices.push_back(FakeDevice::Make(gpu1)); DeviceSet device_set; for (auto& device : devices) device_set.AddDevice(device.get()); AddDevicesToOp(*module_ref, &device_set); auto devices_attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util.cc
"Multiple CPU devices ", device_info_cache.DebugString(devices))); } if (multiple_gpu_devices) { FAILED_TO_PICK_DEVICE(errors::Internal( "Multiple GPU devices ", device_info_cache.DebugString(devices))); } if (multiple_unknown_devices) { FAILED_TO_PICK_DEVICE(errors::Internal( "Multiple unknown devices ", device_info_cache.DebugString(devices))); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.8K bytes - Viewed (0) -
pkg/kubelet/apis/podresources/server_v1alpha1_test.go
pods []*v1.Pod devices []*podresourcesv1.ContainerDevices expectedResponse *v1alpha1.ListPodResourcesResponse }{ { desc: "no pods", pods: []*v1.Pod{}, devices: []*podresourcesv1.ContainerDevices{}, expectedResponse: &v1alpha1.ListPodResourcesResponse{}, }, { desc: "pod without devices", pods: []*v1.Pod{ {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Mar 07 08:12:16 UTC 2024 - 3.9K bytes - Viewed (0) -
pkg/kubelet/cm/devicemanager/checkpoint/checkpoint.go
func New(devEntries []PodDevicesEntry, devices map[string][]string) DeviceManagerCheckpoint { return newV2(devEntries, devices) } func newV2(devEntries []PodDevicesEntry, devices map[string][]string) DeviceManagerCheckpoint { return &Data{ Data: checkpointData{ PodDeviceEntries: devEntries, RegisteredDevices: devices, }, } } // MarshalCheckpoint returns marshalled data
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Mon Apr 15 12:01:56 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_gpu_device.cc
auto device = std::make_unique<XlaDevice>(session_options, options); Status status = device->UseAcceleratorDeviceInfo(); if (!status.ok()) { LOG(INFO) << "Ignoring visible " << DEVICE_GPU_XLA_JIT << " device. Device number is " << i << ", reason: " << status; continue; } devices->push_back(std::move(device)); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_cpu_device.cc
public: Status ListPhysicalDevices(std::vector<string>* devices) override; Status CreateDevices(const SessionOptions& options, const string& name_prefix, std::vector<std::unique_ptr<Device>>* devices) override; }; Status XlaCpuDeviceFactory::ListPhysicalDevices(std::vector<string>* devices) { XlaDeviceFlags* flags = GetXlaDeviceFlags();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/device_util.h
}; // Caches some miscellaneous information about TF devices. Thread compatible. class DeviceInfoCache { public: bool IsGpu(DeviceId device) const { return is_gpu_[device.id()]; } bool IsCpu(DeviceId device) const { return is_cpu_[device.id()]; } absl::string_view GetNameFor(DeviceId device) const { return names_[device.id()]; } absl::StatusOr<DeviceId> GetIdFor(absl::string_view name);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 17:18:31 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 17 07:28:40 UTC 2024 - 9K bytes - Viewed (0) -
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)