[FIX] modified for checking weight grad
authorJiho Chu <jiho.chu@samsung.com>
Tue, 1 Aug 2023 10:34:02 +0000 (19:34 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Wed, 2 Aug 2023 23:25:08 +0000 (08:25 +0900)
commita9518b2e6c56450ff46e47b07f4bd56d7a5b4624
treea1902eccd21171ee5e56cd5b9d1c62672b4f54c8
parentc5a10cda52cb2a87bfccbd8d1d97a07e917cf3a0
[FIX] modified for checking weight grad

This path checks requested memory is weight gradient which information
will be used for planning.

Signed-off-by: Jiho Chu <jiho.chu@samsung.com>
nntrainer/tensor/cache_pool.cpp
nntrainer/tensor/cache_pool.h