Searched refs:in_gpu_mem (Results 1 – 1 of 1) sorted by relevance
156 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeReductionTestCase() local157 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeReductionTestCase()197 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeGatherTestCase() local198 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeGatherTestCase()233 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeBroadcastTestCase() local235 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeBroadcastTestCase()