Searched defs:GetDevicePtr (Results 1 – 2 of 2) sorted by relevance
72 void *TensorRTAllocator::GetDevicePtr(const std::string &tensor_name) { in GetDevicePtr() function in mindspore::lite::TensorRTAllocator
458 void *GetDevicePtr() const { return address_common_->pointer_ref_count_->ptr(); } in GetDevicePtr() function