Searched refs:feed_gpu_fetch_cpu (Results 1 – 1 of 1) sorted by relevance
2420 Session::CallableHandle feed_gpu_fetch_cpu; in TEST() local2423 TF_ASSERT_OK(session->MakeCallable(opts, &feed_gpu_fetch_cpu)); in TEST()2425 TF_ASSERT_OK(session->RunCallable(feed_gpu_fetch_cpu, {gpu_tensor}, in TEST()2427 TF_ASSERT_OK(session->ReleaseCallable(feed_gpu_fetch_cpu)); in TEST()