Searched refs:gpu_ptr (Results 1 – 3 of 3) sorted by relevance
44 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in CheckMask() local47 Status result = exec->SynchronousMemcpyD2H(gpu_ptr, MASK_BYTES, tmp); in CheckMask()66 se::DeviceMemory<int64> gpu_ptr{se::DeviceMemoryBase{ptr, MASK_BYTES}}; in InitMask() local67 Status result = exec->SynchronousMemcpyH2D(mask, MASK_BYTES, &gpu_ptr); in InitMask()
71 se::DeviceMemoryBase gpu_ptr(ptr); in Free()72 stream_exec_->Deallocate(&gpu_ptr); in Free()
382 DeviceMemoryBase gpu_ptr(ptr, num_bytes); in MemoryDebugString() local384 gpu_ptr, num_bytes, gtl::string_as_array(&buf)); in MemoryDebugString()