Searched refs:host_memory_allocate (Results 1 – 5 of 5) sorted by relevance
122 se_.host_memory_allocate = [](const SP_Device* const device, uint64_t size) { in TEST_F()
104 se->host_memory_allocate = HostMemoryAllocate; in PopulateDefaultStreamExecutor()
255 void* (*host_memory_allocate)(const SP_Device* device, uint64_t size); member
119 TF_VALIDATE_NOT_NULL(SP_StreamExecutor, se, host_memory_allocate); in ValidateSPStreamExecutor()256 return stream_executor_->host_memory_allocate(&device_, size); in HostMemoryAllocate()
193 se_.host_memory_allocate = [](const SP_Device* const device, uint64_t size) { in TEST_F()