Searched refs:GPUToCPU (Results 1 – 1 of 1) sorted by relevance
174 Tensor GPUToCPU(const Tensor& device_tensor) { in GPUToCPU() function in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest584 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()