MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / get

Method get

src/tensorrt/impl/tensorrt_engine_cache.cpp:198–204  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

196}
197
198Maybe<TensorRTEngineCache::Engine> TensorRTEngineCacheIO::get(const std::string& key) {
199 MGB_LOCK_GUARD(m_mtx);
200 auto find = m_cache.find(key);
201 if (find == m_cache.end())
202 return None;
203 return find->second;
204}
205
206void TensorRTEngineCacheIO::put(const std::string& key, const Engine& value) {
207 MGB_LOCK_GUARD(m_mtx);

Callers 14

TESTFunction · 0.45
TensorRTOprMethod · 0.45
get_output_var_shapeMethod · 0.45
scn_do_executeMethod · 0.45
TensorRTRuntimeOprMethod · 0.45
get_output_var_shapeMethod · 0.45
scn_do_executeMethod · 0.45
makeMethod · 0.45
ImplMethod · 0.45
update_graphMethod · 0.45
EngineStorageClass · 0.45

Calls 2

findMethod · 0.45
endMethod · 0.45

Tested by 1

TESTFunction · 0.36