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
2929 (*transform_scratch)->mutable_device_memory()->opaque()); in MaybeTransformLayout()4135 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolForward()4140 wsp_mem->mutable_device_memory()->opaque()); in DoPoolForward()4239 pdesc->workspace->mutable_device_memory()->opaque()); in DoPoolBackwardImpl()
2027 DeviceMemory<MAPPED_T>(*(*temp_memory)->mutable_device_memory()); in AllocateStridedBuffer()
2417 a = DeviceMemory<CUDA_T *>(*a_temporary->mutable_device_memory()); in DoBlasInternalImpl()2418 b = DeviceMemory<CUDA_T *>(*b_temporary->mutable_device_memory()); in DoBlasInternalImpl()2419 c = DeviceMemory<CUDA_T *>(*c_temporary->mutable_device_memory()); in DoBlasInternalImpl()