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

Method get_input_tensor

lite/src/network.cpp:139–145  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

137}
138
139std::shared_ptr<Tensor> Network::get_input_tensor(size_t index) {
140 LITE_ERROR_HANDLER_BEGIN
141 LITE_ASSERT(m_loaded, "get_input_tensor should be used after model loaded.");
142 LITE_CHECK_NON_NULL_POINTER(m_impl);
143 return m_impl->get_input_tensor(index);
144 LITE_ERROR_HANDLER_END
145}
146
147std::vector<std::shared_ptr<Tensor>> Network::get_input_tensors(size_t index) {
148 LITE_ERROR_HANDLER_BEGIN

Callers 15

device_inputFunction · 0.45
device_input_outputFunction · 0.45
pinned_host_inputFunction · 0.45
config_user_allocatorFunction · 0.45
basic_load_from_pathFunction · 0.45
basic_load_from_memoryFunction · 0.45
async_forwardFunction · 0.45
set_input_callbackFunction · 0.45
load_from_path_run_cudaFunction · 0.45
reset_inputFunction · 0.45

Calls

no outgoing calls

Tested by 4

TESTFunction · 0.36
test_multi_threadFunction · 0.36
load_device_idFunction · 0.36
mgelite_larFunction · 0.36