Searched refs:CastDeviceMemory (Results 1 – 2 of 2) sorted by relevance
654 DeviceMemory<U> CastDeviceMemory(Tensor* tensor) { in CastDeviceMemory() function1119 reserve_space_data = functor::CastDeviceMemory<uint8, U>( in operator ()()
307 DeviceMemory<U> CastDeviceMemory(Tensor* tensor) { in CastDeviceMemory() function903 CastDeviceMemory<uint8, T>(const_cast<Tensor*>(reserve_space)); in DoBackward()