MCPcopy Create free account
hub / github.com/OAID/Tengine / GetOutputTensor

Method GetOutputTensor

core/lib/graph.cpp:967–983  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

965}
966
967Tensor* Graph::GetOutputTensor(const std::string& name)
968{
969 for(unsigned int i = 0; i < output_nodes.size(); i++)
970 {
971 Node* node = output_nodes[i];
972
973 for(unsigned int j = 0; j < node->GetOutputNum(); j++)
974 {
975 Tensor* tensor = node->GetOutputTensor(j);
976
977 if(tensor->GetName() == name)
978 return tensor;
979 }
980 }
981
982 return nullptr;
983}
984
985bool Graph::Replace(Subgraph* orig_sub, Subgraph* new_sub)
986{

Callers 15

graph_executor.cppFile · 0.45
InferShapeMethod · 0.45
GetInputNodeTensorMethod · 0.45
GetOutputNodeTensorMethod · 0.45
check_graph_inputsFunction · 0.45
do_merge_graphFunction · 0.45
get_node_output_tensorFunction · 0.45
RealCreateFromStaticMethod · 0.45
ReplaceMethod · 0.45
SaveTmNodeMethod · 0.45
SaveTmSubgraphMethod · 0.45
SaveTmNodeMethod · 0.45

Calls 3

GetNameMethod · 0.80
sizeMethod · 0.45
GetOutputNumMethod · 0.45

Tested by 5

test_new_operatorFunction · 0.36
test_new_operatorFunction · 0.36
test_new_operatorFunction · 0.36
test_new_operatorFunction · 0.36
test_new_operatorFunction · 0.36