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

Method get_output_tensor

lite/src/network.cpp:155–161  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

153}
154
155std::shared_ptr<Tensor> Network::get_output_tensor(size_t index) {
156 LITE_ERROR_HANDLER_BEGIN
157 LITE_ASSERT(m_loaded, "get_output_tensor should be used after model loaded.");
158 LITE_CHECK_NON_NULL_POINTER(m_impl);
159 return m_impl->get_output_tensor(index);
160 LITE_ERROR_HANDLER_END
161}
162
163Network& Network::set_async_callback(const AsyncCallback& callback) {
164 LITE_ERROR_HANDLER_BEGIN

Callers 15

device_inputFunction · 0.45
pinned_host_inputFunction · 0.45
config_user_allocatorFunction · 0.45
output_infoFunction · 0.45
output_data_infoFunction · 0.45
basic_load_from_memoryFunction · 0.45
async_forwardFunction · 0.45
set_input_callbackFunction · 0.45
set_output_callbackFunction · 0.45
reset_inputFunction · 0.45
reset_input_outputFunction · 0.45

Calls

no outgoing calls

Tested by 7

TESTFunction · 0.36
TESTFunction · 0.36
test_multi_threadFunction · 0.36
test_output_no_copyFunction · 0.36
test_input_no_copyFunction · 0.36
load_device_idFunction · 0.36
mgelite_larFunction · 0.36