saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update SessionState::GetKernel to just call find() instead of count() and find(). (#1046)

Browse source 
Update SessionState::GetInputNodeInfo to just call find() instead of count() and at().
This commit is contained in:
Scott McKay 2019-05-16 10:02:01 -07:00 committed by GitHub
parent 2e19b14e4e
commit 6c65640708
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
onnxruntime/core/framework/session_state.cc
View file

@ -21,11 +21,8 @@ void SessionState::SetGraphViewer(std::unique_ptr<onnxruntime::GraphViewer> grap
const GraphViewer* SessionState::GetGraphViewer() const { return graph_viewer_.get(); }
const OpKernel* SessionState::GetKernel(NodeIndex node_id) const {
if (session_kernels_.count(node_id) == 0) {
return nullptr;
}
return session_kernels_.find(node_id)->second.get();
auto kernel = session_kernels_.find(node_id);
return (kernel != session_kernels_.cend()) ? kernel->second.get() : nullptr;
}
void SessionState::AddKernel(onnxruntime::NodeIndex node_id, std::unique_ptr<OpKernel> p_kernel) {
@ -148,10 +145,12 @@ common::Status SessionState::AddInputNameToNodeInfoMapping(const std::string& in
common::Status SessionState::GetInputNodeInfo(const std::string& input_name,
std::vector<NodeInfo>& node_info_vec) const {
if (!input_names_to_nodeinfo_mapping_.count(input_name)) {
auto entry = input_names_to_nodeinfo_mapping_.find(input_name);
if (entry == input_names_to_nodeinfo_mapping_.cend()) {
return Status(ONNXRUNTIME, FAIL, "Failed to find input name in the mapping: " + input_name);
}
node_info_vec = input_names_to_nodeinfo_mapping_.at(input_name);
node_info_vec = entry->second;
return Status::OK();
}