Searched refs:IsCUDATensor (Results 1 – 2 of 2) sorted by relevance
540 bool IsCUDATensor(const Tensor& t) { in IsCUDATensor() function580 EXPECT_FALSE(IsCUDATensor(y_cpu)); in TestTwoDeviceMult()583 EXPECT_TRUE(IsCUDATensor(y_gpu)); in TestTwoDeviceMult()612 EXPECT_TRUE(IsCUDATensor(y1)); in TestTwoDeviceInputOutput()615 EXPECT_FALSE(IsCUDATensor(y1)); in TestTwoDeviceInputOutput()620 EXPECT_TRUE(IsCUDATensor(y2)); in TestTwoDeviceInputOutput()623 EXPECT_FALSE(IsCUDATensor(y2)); in TestTwoDeviceInputOutput()757 EXPECT_FALSE(IsCUDATensor(y1)); in TEST_F()760 EXPECT_TRUE(IsCUDATensor(y2)); in TEST_F()930 EXPECT_FALSE(IsCUDATensor(returned_handle)); in TEST_F()[all …]
2324 bool IsCUDATensor(const Tensor& t) { in IsCUDATensor() function2383 ASSERT_TRUE(IsCUDATensor(gpu_tensor)); in TEST()2457 ASSERT_TRUE(IsCUDATensor(gpu_tensor)) << DataType_Name(dtype); in TestFeedAndFetchTensorsInDeviceMemory()