Searched refs:y_gpu (Results 1 – 1 of 1) sorted by relevance
568 Tensor y_gpu; in TestTwoDeviceMult() local570 inst_opts, {x}, {&y_cpu, &y_gpu}); in TestTwoDeviceMult()583 EXPECT_TRUE(IsCUDATensor(y_gpu)); in TestTwoDeviceMult()584 Tensor y_gpu_on_cpu = fixture->GPUToCPU(y_gpu); in TestTwoDeviceMult()