Home
last modified time | relevance | path

Searched refs:CreateGPUDevice (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.h300 Status CreateGPUDevice(const SessionOptions& options,
305 virtual std::unique_ptr<BaseGPUDevice> CreateGPUDevice(
Dgpu_device_factory.cc63 std::unique_ptr<BaseGPUDevice> CreateGPUDevice( in CreateGPUDevice() function in tensorflow::GPUDeviceFactory
Dgpu_device.cc1155 TF_RETURN_IF_ERROR(CreateGPUDevice(options, name_prefix, tf_gpu_id, bytes, in CreateDevices()
1182 Status BaseGPUDeviceFactory::CreateGPUDevice( in CreateGPUDevice() function in tensorflow::BaseGPUDeviceFactory
1218 std::unique_ptr<BaseGPUDevice> gpu_device = CreateGPUDevice( in CreateGPUDevice()