Searched refs:y_gpu_on_cpu (Results 1 – 1 of 1) sorted by relevance
584 Tensor y_gpu_on_cpu = fixture->GPUToCPU(y_gpu); in TestTwoDeviceMult() local585 test::ExpectTensorEqual<float>(y_gpu_on_cpu, in TestTwoDeviceMult()