Searched refs:y_cpu (Results 1 – 1 of 1) sorted by relevance
567 Tensor y_cpu; in TestTwoDeviceMult() local570 inst_opts, {x}, {&y_cpu, &y_gpu}); in TestTwoDeviceMult()580 EXPECT_FALSE(IsCUDATensor(y_cpu)); in TestTwoDeviceMult()581 test::ExpectTensorEqual<float>(y_cpu, test::AsTensor<float>({2, 4, 6})); in TestTwoDeviceMult()