Searched refs:out_gpu_mem (Results 1 – 1 of 1) sorted by relevance
269 auto out_gpu_mem = AsDeviceMemory(out_gpu.flat<Scalar>().data()); in VerifyResults() local270 stream->ThenMemcpy(out_cpu.flat<Scalar>().data(), out_gpu_mem, in VerifyResults()