Searched refs:trt_tensors_ (Results 1 – 2 of 2) sorted by relevance
638 std::unordered_map<string, TRT_TensorOrWeights> trt_tensors_; variable
1566 if (trt_tensors_.insert({name, std::move(input)}).second) return Status::OK(); in AddTensorOrWeights()1572 if (!trt_tensors_.count(name)) { in GetTensorOrWeights()1576 *output = trt_tensors_.at(name); in GetTensorOrWeights()2029 if (trt_tensors_.count(name)) { in GetInputs()2030 TRT_TensorOrWeights input = trt_tensors_.at(name); in GetInputs()