- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for memory_space (0.16 sec)
-
tensorflow/compiler/jit/xla_tensor.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
} absl::Status Init() override { return absl::OkStatus(); } DeviceMemoryBase Allocate(uint64 size, int64_t memory_space) override { SP_DeviceMemoryBase mem = {SP_DEVICE_MEMORY_BASE_STRUCT_SIZE}; stream_executor_->allocate(&device_, size, memory_space, &mem); absl::Status status = ValidateSPDeviceMemoryBase(mem); if (!status.ok()) { LOG(ERROR) << status.message(); }
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_util.cc
namespace stream_executor { namespace test_util { /*** Functions for creating SP_StreamExecutor ***/ void Allocate(const SP_Device* const device, uint64_t size, int64_t memory_space, SP_DeviceMemoryBase* const mem) {} void Deallocate(const SP_Device* const device, SP_DeviceMemoryBase* const mem) { } void* HostMemoryAllocate(const SP_Device* const device, uint64_t size) { return nullptr; }
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.h
// `SP_DeviceMemoryBase` representing that allocation. In the case of failure, // nullptr is returned. // `memory_space` is reserved for a potential future usage and should be set // to 0. void (*allocate)(const SP_Device* device, uint64_t size, int64_t memory_space, SP_DeviceMemoryBase* mem); // Deallocate the device memory previously allocated via this interface.
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/c/experimental/stream_executor/stream_executor_test.cc
SP_TimerFns timer_fns_; std::unique_ptr<CPlatform> cplatform_; }; TEST_F(StreamExecutorTest, Allocate) { se_.allocate = [](const SP_Device* const device, uint64_t size, int64_t memory_space, SP_DeviceMemoryBase* const mem) { mem->struct_size = SP_DEVICE_MEMORY_BASE_STRUCT_SIZE; mem->opaque = malloc(size); mem->size = size; }; se_.deallocate = [](const SP_Device* const device,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 19:54:04 UTC 2024 - 26.5K bytes - Viewed (0)