Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for DeviceMemoryUsage (0.21 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top