projects / platform / upstream / nnstreamer.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 1356c2a) | patch
[filter] Bug fix for pytorch filter with gpu
authorParichay Kapoor <pk.kapoor@samsung.com>
Thu, 27 Jan 2022 08:44:04 +0000 (17:44 +0900)
committerjaeyun-jung <39614140+jaeyun-jung@users.noreply.github.com>
Fri, 4 Feb 2022 06:29:38 +0000 (15:29 +0900)
commit844a557c0bd612d421028bcf93f8712c09ab3cd0
tree50ec1161cccf8dbd57c24da12156f5b3bf0dadcatree | snapshot
parent1356c2a43cd25fb94ee71c214bce40bc65f22f30commit | diff
[filter] Bug fix for pytorch filter with gpu

ptyorch filter when working with gpu backend had bug when converting
between the devices. The tensors returned by the conversion were not
catched. This patch provides the corresponding fix.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
ext/nnstreamer/tensor_filter/tensor_filter_pytorch.cc diff | blob | history
Domain: Machine Learning / ML Framework;