Searched refs:gpu_helper_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 177 GPUDeviceTestHelper* gpu_helper_; member in tensorflow::__anon66d29a500511::EMBenchmarkHelper
204 EMBenchmarkHelper(GPUDeviceTestHelper* h) : gpu_helper_(h) {} in EMBenchmarkHelper()
209 gpu_inputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
214 gpu_outputs_.push_back(Tensor(gpu_helper_->gpu_allocator(), DT_FLOAT, in ReInit()
220 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()
244 return CreateOpKernel("GPU", gpu_helper_->gpu(), in GetOpKernel()
245 gpu_helper_->gpu_allocator(), node_def, in GetOpKernel()
282 params->device = gpu_helper_->gpu(); in PrepOpKernel()
298 params->resource_manager = gpu_helper_->gpu()->resource_manager(); in PrepOpKernel()
[all …]
|