Searched refs:GetDevicePtr (Results 1 – 2 of 2) sorted by relevance
42 void *GetDevicePtr(const std::string &tensor_name);
74 void *TensorRTAllocator::GetDevicePtr(const std::string &tensor_name) { in GetDevicePtr() function in mindspore::lite::TensorRTAllocator