Searched refs:host_allocator (Results 1 – 1 of 1) sorted by relevance
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()