Home
last modified time | relevance | path

Searched refs:unified_memory_allocate (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc122 TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, unified_memory_allocate); in ValidateSPStreamExecutor()
267 CHECK(stream_executor_->unified_memory_allocate); in UnifiedMemoryAllocate()
268 return stream_executor_->unified_memory_allocate(&device_, size); in UnifiedMemoryAllocate()
H A Dstream_executor.h263 void* (*unified_memory_allocate)(const SP_Device* device, uint64_t bytes); member
H A Dstream_executor_test.cc214 se_.unified_memory_allocate = [](const SP_Device* const device, in TEST_F()