Searched defs:gpu_src_ptr (Results 1 – 1 of 1) sorted by relevance
161 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in SetProtoFromGPU() local215 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in DeviceToDeviceCopy() local282 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local440 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToSameGPU() local