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_test.cc414 string gpu_device_name; in TensorHandleSilentCopy() local
486 string gpu_device_name; in SetAndGetOpDevices() local
564 string gpu_device_name; in TEST() local
615 std::string gpu_device_name; in ExecuteAdd() local
Dc_api_experimental_test.cc448 string gpu_device_name; in TEST() local
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc2363 const string gpu_device_name = GPUDeviceName(session.get()); in TEST() local
2435 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemory() local
2491 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.py152 def gpu_device_name(): function
/external/tensorflow/tensorflow/c/
Dc_api_test.cc109 const string gpu_device_name = GPUDeviceName(sess); in GPUDeviceName() local