dropout_op.cc
2.12 KB
-
Add support to query the size for the allocated memory · 58c5371eSummary: This commit adds the ``memory_allocated`` API for ``dragon.Workspace`` to query the size of allocated memory (and optionally on a specified device).
Ting PAN committed