Commit Graph

9 Commits (a7bb983343a8fe9d8518a0b31388b67faf3f9320)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
8 years ago
wanghaoshuang 1fdad1a60a Update transform invocation
8 years ago
wanghaoshuang 2cde56c5a5 Use Transform instead of eigen
8 years ago
wanghaoshuang 743dfd82e7 Add nullptr check
8 years ago
wanghaoshuang 14fb15b685 Remove const cast for device context
8 years ago
wanghaoshuang a345b7195e 1. Add CUDA stream when launching kernel.
8 years ago
wanghaoshuang 2321a37b3c fix issues
8 years ago
wanghaoshuang 6e964ad5cb Fix issues
8 years ago
wanghaoshuang 987cdf1168 Add clip op
8 years ago