- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for DeviceMemoryUsage (0.21 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor_test_util.cc
} void HostMemoryDeallocate(const SP_Device* const device, void* mem) {} TF_Bool GetAllocatorStats(const SP_Device* const device, SP_AllocatorStats* const stats) { return true; } TF_Bool DeviceMemoryUsage(const SP_Device* const device, int64_t* const free, int64_t* const total) { return true; } void CreateStream(const SP_Device* const device, SP_Stream* stream,
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_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.cc
return tsl::errors::Unimplemented( "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)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0)