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

Method get_io_tensor

lite/src/network.cpp:121–128  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

119}
120
121std::shared_ptr<Tensor> Network::get_io_tensor(
122 std::string name, LiteTensorPhase phase) {
123 LITE_ERROR_HANDLER_BEGIN
124 LITE_ASSERT(m_loaded, "get_io_tensor should be used after model loaded.");
125 LITE_CHECK_NON_NULL_POINTER(m_impl);
126 return m_impl->get_io_tensor(name, phase);
127 LITE_ERROR_HANDLER_END
128}
129
130std::vector<std::shared_ptr<Tensor>> Network::get_discrete_tensors(
131 std::string name, LiteTensorPhase phase) {

Callers 11

device_input_outputFunction · 0.45
detect_yoloxFunction · 0.45
TESTFunction · 0.45
TESTFunction · 0.45
test_output_no_copyFunction · 0.45
test_input_no_copyFunction · 0.45
test_io_no_copy_axFunction · 0.45
format_and_printMethod · 0.45
get_io_infoMethod · 0.45
LITE_get_io_tensorFunction · 0.45

Calls

no outgoing calls

Tested by 5

TESTFunction · 0.36
TESTFunction · 0.36
test_output_no_copyFunction · 0.36
test_input_no_copyFunction · 0.36
test_io_no_copy_axFunction · 0.36