Searched refs:CastDeviceMemory (Results 1 – 1 of 1) sorted by relevance
307 DeviceMemory<U> CastDeviceMemory(Tensor* tensor) { in CastDeviceMemory() function843 CastDeviceMemory<uint8, T>(const_cast<Tensor*>(reserve_space)); in DoBackward()