Searched refs:host_allocator (Results 1 – 5 of 5) sorted by relevance
37 std::unique_ptr<::tfrt::HostAllocator> host_allocator = in GetStaticHostContext() local40 std::move(host_allocator), in GetStaticHostContext()
56 std::unique_ptr<HostAllocator> host_allocator = CreateMallocAllocator(); in SetUp() local58 std::move(host_allocator), in SetUp()
172 auto& host_allocator = *at::getCPUAllocator(); in apply_ldl_solve_cusolver()173 auto workdata_host = host_allocator.allocate(worksize_host); in apply_ldl_solve_cusolver()743 auto& host_allocator = *at::getCPUAllocator(); in apply_cholesky_cusolver_potrf_looped() local744 auto workdata_host = host_allocator.allocate(worksize_host * batch_size); in apply_cholesky_cusolver_potrf_looped()1016 auto& host_allocator = *at::getCPUAllocator(); in apply_geqrf() local1017 auto work_host_data = host_allocator.allocate(worksize_host); in apply_geqrf()1266 auto& host_allocator = *at::getCPUAllocator(); in apply_syevd() local1267 auto work_host_data = host_allocator.allocate(worksize_host); in apply_syevd()
22 #include "tfrt/host_context/host_allocator.h" // from @tf_runtime
158 Allocator* host_allocator() { return host_allocator_; } in host_allocator() function in tensorflow::GPUDeviceTestHelper220 host_inputs_.push_back(Tensor(gpu_helper_->host_allocator(), DT_FLOAT, in ReInit()229 host_outputs_.push_back(Tensor(gpu_helper_->host_allocator(), DT_FLOAT, in ReInit()