Searched refs:trt_tensors_ (Results 1 – 2 of 2) sorted by relevance
544 std::unordered_map<string, TRT_TensorOrWeights> trt_tensors_; variable
1140 if (trt_tensors_.insert({name, std::move(input)}).second) return Status::OK(); in AddTensorOrWeights()1146 if (!trt_tensors_.count(name)) { in GetTensorOrWeights()1150 *output = trt_tensors_.at(name); in GetTensorOrWeights()1405 if (trt_tensors_.count(name)) { in GetInputs()1406 TRT_TensorOrWeights input = trt_tensors_.at(name); in GetInputs()