Searched refs:gpu_device_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime_test.cc | 126 "/job:a/replica:0/task:0/device:GPU:0", &gpu_device_); in ProcessFunctionLibraryRuntimeTest()
128 CHECK_EQ(nullptr, gpu_device_); in ProcessFunctionLibraryRuntimeTest()
176 CHECK(gpu_device_); in GPUToCPU()
177 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in GPUToCPU()
179 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()
183 ->CopyDeviceTensorToCPUSync(&device_tensor, "", gpu_device_, in GPUToCPU()
194 CHECK(gpu_device_); in CPUToGPU()
195 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in CPUToGPU()
197 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
199 Tensor device_tensor(gpu_device_->GetAllocator({}), cpu_tensor.dtype(), in CPUToGPU()
[all …]
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_device.cu | 70 …ensorMap<Eigen::Tensor<float, 3> >& out) : in1_(in1), in2_(in2), out_(out), gpu_device_(&stream_) { in GPUContext()
89 const Eigen::GpuDevice& device() const { return gpu_device_; } in device()
108 Eigen::GpuDevice gpu_device_; member
|