Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc154 VALIDATE_MEMBER(SP_StreamExecutor, se, unified_memory_allocate); in ValidateSPStreamExecutor()
295 CHECK(stream_executor_->unified_memory_allocate); in UnifiedMemoryAllocate()
296 return stream_executor_->unified_memory_allocate(&device_, size); in UnifiedMemoryAllocate()
Dstream_executor.h263 void* (*unified_memory_allocate)(const SP_Device* device, uint64_t bytes); member
Dstream_executor_test.cc390 se_.unified_memory_allocate = [](const SP_Device* const device, in TEST_F()