- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for device_memory_usage (0.29 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
se->deallocate = Deallocate; se->host_memory_allocate = HostMemoryAllocate; se->host_memory_deallocate = HostMemoryDeallocate; se->get_allocator_stats = GetAllocatorStats; se->device_memory_usage = DeviceMemoryUsage; se->create_stream = CreateStream; se->destroy_stream = DestroyStream; se->create_stream_dependency = CreateStreamDependency; se->get_stream_status = GetStreamStatus;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 21 20:09:00 UTC 2023 - 8.7K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
"EnablePeerAccessTo is not supported by pluggable device."); } bool CanEnablePeerAccessTo(StreamExecutor* other) override { return false; } bool DeviceMemoryUsage(int64_t* free, int64_t* total) const override { return stream_executor_->device_memory_usage( &device_, reinterpret_cast<int64_t*>(free), reinterpret_cast<int64_t*>(total)); } // Creates a new DeviceDescription object.
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_test.cc
ASSERT_TRUE(optional_stats.has_value()); AllocatorStats stats = optional_stats.value(); ASSERT_EQ(stats.bytes_in_use, 123); } TEST_F(StreamExecutorTest, DeviceMemoryUsage) { se_.device_memory_usage = [](const SP_Device* const device, int64_t* const free, int64_t* const total) -> TF_Bool { *free = 45; *total = 7;
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.h
// Fills the underlying device memory usage information, if it is // available. If it is not available (false is returned), free/total need not // be initialized. TF_Bool (*device_memory_usage)(const SP_Device* device, int64_t* free, int64_t* total); /*** STREAM CALLBACKS ***/ // Creates SP_Stream. This call should also allocate stream
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 24 08:40:35 UTC 2022 - 21.6K bytes - Viewed (0)