Home
last modified time | relevance | path

Searched defs:gpu_device_name (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc452 string gpu_device_name; in TEST() local
Dc_api_test.cc415 string gpu_device_name; in TensorHandleSilentCopy() local
487 string gpu_device_name; in SetAndGetOpDevices() local
565 string gpu_device_name; in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc2389 const string gpu_device_name = GPUDeviceName(session.get()); in TEST() local
2461 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemory() local
2517 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable() local
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_kernels.cc223 auto gpu_device_name = absl::StrCat(gpu_device_type, ":", gpu_ordinal); in InjectTfGpuResourcesHelper() local
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py150 def gpu_device_name(): function
/external/tensorflow/tensorflow/c/
Dc_api_test.cc110 const string gpu_device_name = GPUDeviceName(sess); in GPUDeviceName() local