Searched refs:mutable_device_memory (Results 1 – 5 of 5) sorted by relevance
62 DeviceMemoryBase* mutable_device_memory();112 DeviceMemory<T>* mutable_device_memory() { in mutable_device_memory() function115 TemporaryDeviceMemoryBase::mutable_device_memory()); in mutable_device_memory()
28 DeviceMemoryBase* TemporaryDeviceMemoryBase::mutable_device_memory() { in mutable_device_memory() function in stream_executor::TemporaryDeviceMemoryBase
1240 a = DeviceMemory<CUDA_T *>(*a_temporary->mutable_device_memory()); in DoBlasInternalImpl()1241 b = DeviceMemory<CUDA_T *>(*b_temporary->mutable_device_memory()); in DoBlasInternalImpl()1242 c = DeviceMemory<CUDA_T *>(*c_temporary->mutable_device_memory()); in DoBlasInternalImpl()
4065 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolForward()4069 wsp_mem->mutable_device_memory()->opaque()); in DoPoolForward()4221 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolBackward()
753 DeviceMemory<MAPPED_T>(*(*temp_memory)->mutable_device_memory()); in AllocateStridedBuffer()