Searched defs:in_gpu_mem (Results 1 – 1 of 1) sorted by relevance
156 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeReductionTestCase() local197 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeGatherTestCase() local233 auto in_gpu_mem = AsDeviceMemory(in_gpu.flat<Scalar>().data()); in MakeBroadcastTestCase() local