Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime_test.cc568 Tensor y_gpu; in TestTwoDeviceMult() local
570 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()