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.h354 Status CreateGPUDevice(const SessionOptions& options,
360 virtual std::unique_ptr<BaseGPUDevice> CreateGPUDevice(
Dgpu_device_factory.cc64 std::unique_ptr<BaseGPUDevice> CreateGPUDevice( in CreateGPUDevice() function in tensorflow::GPUDeviceFactory
Dgpu_device.cc1373 TF_RETURN_IF_ERROR(CreateGPUDevice(options, name_prefix, tf_gpu_id, bytes, in CreateDevices()
1401 Status BaseGPUDeviceFactory::CreateGPUDevice( in CreateGPUDevice() function in tensorflow::BaseGPUDeviceFactory
1452 std::unique_ptr<BaseGPUDevice> gpu_device = CreateGPUDevice( in CreateGPUDevice()