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

Method get_discrete_tensors

lite/src/network.cpp:130–137  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

128}
129
130std::vector<std::shared_ptr<Tensor>> Network::get_discrete_tensors(
131 std::string name, LiteTensorPhase phase) {
132 LITE_ERROR_HANDLER_BEGIN
133 LITE_ASSERT(m_loaded, "get_io_tensor should be used after model loaded.");
134 LITE_CHECK_NON_NULL_POINTER(m_impl);
135 return m_impl->get_discrete_tensors(name, phase);
136 LITE_ERROR_HANDLER_END
137}
138
139std::shared_ptr<Tensor> Network::get_input_tensor(size_t index) {
140 LITE_ERROR_HANDLER_BEGIN

Callers 2

TESTFunction · 0.45
LITE_get_discrete_tensorFunction · 0.45

Calls

no outgoing calls

Tested by 1

TESTFunction · 0.36