Searched refs:trt_in_tensor_name_ (Results 1 – 2 of 2) sorted by relevance
326 trt_in_tensor_name_.push_back(tensor.Name()); in Prepare()358 for (size_t i = 0; i < trt_in_tensor_name_.size(); i++) { in ReSize()361 … if (std::strcmp(this->network_->getInput(j)->getName(), trt_in_tensor_name_[i].c_str()) != 0) { in ReSize()389 …auto device_ptr = runtime_->GetAllocator()->MallocDeviceMem(trt_in_tensor_name_[i], inputs_[i].Dat… in ReSize()395 int index = this->engine_->getBindingIndex(trt_in_tensor_name_[i].c_str()); in ReSize()400 …MS_LOG(DEBUG) << "in tensor " << trt_in_tensor_name_[i] << " dims at " << od << " is " << input_di… in ReSize()433 if (runtime_->GetAllocator()->GetMemIsValid(trt_in_tensor_name_[i])) { in Execute()434 MS_LOG(INFO) << "no need memcpy to cuda for input tensor: " << trt_in_tensor_name_[i]; in Execute()437 …int ret = runtime_->GetAllocator()->SyncMemInHostAndDevice(inputs_[i], trt_in_tensor_name_[i], tru… in Execute()439 MS_LOG(ERROR) << "sync mem from host to device failed for " << trt_in_tensor_name_[i]; in Execute()[all …]
92 std::vector<std::string> trt_in_tensor_name_; variable