Searched defs:MallocDeviceMem (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ | ||
| D | tensorrt_allocator.cc | 25 void *TensorRTAllocator::MallocDeviceMem(const TensorInfo &host_tensor, size_t size) { in MallocDeviceMem() function in mindspore::lite::TensorRTAllocator |