Searched refs:CPUToGPU (Results 1 – 1 of 1) sorted by relevance
192 Tensor CPUToGPU(const Tensor& cpu_tensor) { in CPUToGPU() function in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest600 x1 = fixture->CPUToGPU(x1); in TestTwoDeviceInputOutput()604 x2 = fixture->CPUToGPU(x2); in TestTwoDeviceInputOutput()750 Tensor x1 = CPUToGPU(test::AsTensor<float>({1, 2})); in TEST_F()912 Tensor resource_value = CPUToGPU(test::AsTensor<float>({10, 20})); in TEST_F()922 Tensor x1 = CPUToGPU(test::AsTensor<float>({1, 2})); in TEST_F()1043 Tensor resource_value = CPUToGPU(test::AsScalar<int>(10)); in TEST_F()