Searched defs:gpu_src_ptr (Results 1 – 2 of 2) sorted by relevance
160 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in SetProtoFromGPU() local214 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in DeviceToDeviceCopy() local293 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToCPU() local486 DeviceMemoryBase gpu_src_ptr(src_ptr, total_bytes); in CopyGPUTensorToSameGPU() local
419 se::DeviceMemoryBase gpu_src_ptr(DMAHelper::base(ctx->mutable_output(0)), in DoAddChain() local