- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for device_vendor (0.15 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test.cc
SE_CreateDeviceParams* params, TF_Status* status) { params->device->hardware_name = hardware_name; params->device->device_vendor = vendor; params->device->pci_bus_id = pci_bus_id; }; device_fns_.get_numa_node = [](const SP_Device* device) { return 123; }; device_fns_.get_memory_bandwidth = [](const SP_Device* device) -> int64_t {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
internal::DeviceDescriptionBuilder builder; if (device_.hardware_name != nullptr) { builder.set_name(device_.hardware_name); } if (device_.device_vendor != nullptr) { builder.set_device_vendor(device_.device_vendor); } if (device_.pci_bus_id != nullptr) { builder.set_pci_bus_id(device_.pci_bus_id); } if (device_fns_->get_numa_node != nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.h
// Must be null-terminated. const char* hardware_name; // [Optional] // Device vendor name. Used for printing. // Must be null-terminated. const char* device_vendor; // [Optional] // Returns the PCI bus identifier for this device, of the form // [domain]:[bus]:[device].[function] // where domain number is usually 0000. // Example: 0000:00:02.1
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_device_context.h
// Copies 'cpu_tensor' to `device_memory_base_` with `shape_`. // `device_tensor` is unused. void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device, Tensor* device_tensor, StatusCallback done, bool sync_dst_compute) const override; void CopyDeviceTensorToCPU(const Tensor* device_tensor, StringPiece tensor_name, Device* device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_send_recv_device_context_test.cc
TEST_F(XlaHostSendRecvDeviceContextTest, CopyDeviceTensorToCPU) { SetDevice("GPU"); Tensor origin_cpu_tensor(host_allocator_, DT_FLOAT, TensorShape({2, 2})); test::FillValues<float>(&origin_cpu_tensor, {1.2, 2.3, 3.4, 4.5}); Tensor device_tensor(device_allocator_, DT_FLOAT, TensorShape({2, 2})); Tensor dest_cpu_tensor(host_allocator_, DT_FLOAT, TensorShape({2, 2})); stream_executor::Platform* platform =
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/compiler/jit/xla_device_context.cc
device_tensor->shape(), device_tensor->dtype(), std::nullopt); Status status = [&]() -> Status { TF_ASSIGN_OR_RETURN(xla::Shape shape, shape_determination_fns_.shape_representation_fn( device_tensor->shape(), device_tensor->dtype(), /*fast_mem=*/false, layout_preference)); // The device tensor should always be fresh.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_device_context.cc
// tensor. absl::StatusOr<Tensor> t = MakeTensorFromPjRtBuffer( device_tensor->dtype(), device_tensor->shape(), std::move(*buffer_or)); if (!t.ok()) { done(t.status()); return; } *device_tensor = *t; } else { AsyncValueTensor* result_tensor = tensorflow::AsyncValueTensor::FromTensor(device_tensor); // The result tensor should be newly allocated, which does not point to a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 08:49:31 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_host_recv_device_context.h
Tensor* device_tensor, StatusCallback done, bool sync_dst_compute) const override { done(errors::Internal("host->device copy not implemented.")); } // Copies `device_memory_base_` with `shape_` into `cpu_tensor`. // `device_tensor` is unused. void CopyDeviceTensorToCPU(const Tensor* device_tensor,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/jit/device_context_test.cc
TF_ASSERT_OK(device_context_->CopyCPUTensorToDeviceSync( &origin_cpu_tensor, device_.get(), &device_tensor)); TF_ASSERT_OK(device_context_->CopyDeviceTensorToCPUSync( &device_tensor, "", device_.get(), &dest_cpu_tensor)); LOG(INFO) << "H2D - D2H roundtrip completes. tensor: " << dest_cpu_tensor.DebugString(4); tensorflow::test::ExpectClose(origin_cpu_tensor, dest_cpu_tensor); } #endif
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/jit/pjrt_device_context.h
void CopyCPUTensorToDevice(const Tensor* cpu_tensor, Device* device, Tensor* device_tensor, StatusCallback done, bool sync_dst_compute) const override; void CopyDeviceTensorToCPU(const Tensor* device_tensor, absl::string_view tensor_name, Device* device, Tensor* cpu_tensor, StatusCallback done) override;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jul 19 19:27:39 UTC 2023 - 2.7K bytes - Viewed (0)