Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc174 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU() function in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest
584 Tensor y_gpu_on_cpu = fixture->GPUToCPU(y_gpu); in TestTwoDeviceMult()
613 y1 = fixture->GPUToCPU(y1); in TestTwoDeviceInputOutput()
621 y2 = fixture->GPUToCPU(y2); in TestTwoDeviceInputOutput()
761 y2 = GPUToCPU(y2); in TEST_F()
932 y2 = GPUToCPU(y2); in TEST_F()
942 read_resource = GPUToCPU(read_resource); in TEST_F()