Commit Graph

7 Commits (74b122889cbce2aa3add92784d0b4a621abfdf45)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
Yang Yu 3187451ae7
CompareOp's kernel device type is decided by input tensor place
7 years ago
dangqingqing 41a2321a0e Refine platform::Transform function and fix prelu_op testing.
8 years ago
Yu Yang 87e4e25db1 Change Transform API
8 years ago
Yu Yang f8c6792aa3 Extract DevPtrCast to device_ptr_cast.h
8 years ago
Yu Yang 6fbf097bcc Mark thrust::device_ptr in transform
8 years ago
Yu Yang c5fa417c62 Host and device transform API
8 years ago