MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / engine_get_location

Function engine_get_location

python/src/infer/pyCore.cpp:198–201  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

196};
197
198TensorLocation engine_get_location(ICudaEngine& self, std::string const& name)
199{
200 return self.getLocation(self.getBindingIndex(name.c_str()));
201};
202
203// TODO: Add slicing support?
204static const auto engine_getitem = [](ICudaEngine& self, int32_t pyIndex) {

Callers

nothing calls this directly

Calls 3

getBindingIndexMethod · 0.80
c_strMethod · 0.80
getLocationMethod · 0.45

Tested by

no test coverage detected