Searched refs:mutable_device_memory (Results 1 – 4 of 4) 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
2197 a = DeviceMemory<CUDA_T *>(*a_temporary->mutable_device_memory()); in DoBlasGemmBatchedInternal()2198 b = DeviceMemory<CUDA_T *>(*b_temporary->mutable_device_memory()); in DoBlasGemmBatchedInternal()2199 c = DeviceMemory<CUDA_T *>(*c_temporary->mutable_device_memory()); in DoBlasGemmBatchedInternal()
2800 (*transform_scratch)->mutable_device_memory()->opaque()); in MaybeTransformLayout()