Commit Graph

7 Commits (9362d85e0ef9afb0fcd36e12d0a4eac92f08265f)

Author SHA1 Message Date
Chen Weihang aa0f254fbe
Add macro BOOST_GET to enrich the error information of boost :: get (#24175)
5 years ago
Huihuang Zheng 12542320c5
Replace TemporaryAllocator by CUDADeviceContextAllocator (#18989)
6 years ago
Yiqun Liu 7e463c84a6
Optimize the concat and split cuda implementation for cases when the number of inputs/outputs is less than 5. (#17979)
6 years ago
Yiqun Liu 5782dddad0
Optimize the concat and split kernel for specical cases when the number of inputs/outputs is 2 (#17415)
6 years ago
chengduo b9fb03cf54
Move GetTensor to tensor_util (#15011)
6 years ago
chengduo 79bd6dfa18
[Feature] Add Temporary Allocator (#14875)
7 years ago
chengduo a7497653d0
Refine Split op (#13967)
7 years ago