Home
last modified time | relevance | path

Searched refs:CPUToGPU (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc192 Tensor CPUToGPU(const Tensor& cpu_tensor) { in CPUToGPU() function in tensorflow::__anon96a4e55d0111::ProcessFunctionLibraryRuntimeTest
600 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()