Commit Graph

6 Commits (09866fb75f8522e0cea56ccc40fee76cdf7d6be7)

Author SHA1 Message Date
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