Searched refs:MarkMemValid (Results 1 – 3 of 3) sorted by relevance
442 runtime_->GetAllocator()->MarkMemValid(trt_in_tensor_name_[i], true); in Execute()469 runtime_->GetAllocator()->MarkMemValid(trt_out_tensor_name_[i], true); in Execute()475 runtime_->GetAllocator()->MarkMemValid(trt_out_tensor_name_[i], false); in Execute()479 runtime_->GetAllocator()->MarkMemValid(trt_in_tensor_name_[i], false); in Execute()
52 void MarkMemValid(const std::string &name, bool isValid);
61 void TensorRTAllocator::MarkMemValid(const std::string &name, bool isValid) { in MarkMemValid() function in mindspore::lite::TensorRTAllocator