Searched defs:host_allocator (Results 1 – 4 of 4) sorted by relevance
37 std::unique_ptr<::tfrt::HostAllocator> host_allocator = in GetStaticHostContext() local
743 auto& host_allocator = *at::getCPUAllocator(); in apply_cholesky_cusolver_potrf_looped() local1016 auto& host_allocator = *at::getCPUAllocator(); in apply_geqrf() local1266 auto& host_allocator = *at::getCPUAllocator(); in apply_syevd() local
56 std::unique_ptr<HostAllocator> host_allocator = CreateMallocAllocator(); in SetUp() local
158 Allocator* host_allocator() { return host_allocator_; } in host_allocator() function in tensorflow::GPUDeviceTestHelper