HomeSort by relevance Sort by last modified time
    Searched refs:CUDAHostAllocator (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
pool_allocator.h 181 class CUDAHostAllocator : public SubAllocator {
184 explicit CUDAHostAllocator(perftools::gputools::StreamExecutor* stream_exec)
188 ~CUDAHostAllocator() override {}
211 TF_DISALLOW_COPY_AND_ASSIGN(CUDAHostAllocator);
pool_allocator_test.cc 33 new CUDAHostAllocator(
51 new CUDAHostAllocator(
84 new CUDAHostAllocator(
125 TEST(PoolAllocatorTest, CudaHostAllocator) {
130 new CUDAHostAllocator(
207 new CUDAHostAllocator(
process_state.cc 280 new BFCAllocator(new CUDAHostAllocator(se), cuda_host_mem_limit,

Completed in 2022 milliseconds