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

Method RemoveTensor

core/lib/graph.cpp:365–402  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

363}
364
365bool Graph::RemoveTensor(Tensor* tensor)
366{
367 /* remove all refers from consumer */
368
369 for(unsigned int i = 0; i < tensor->consumer.size(); i++)
370 {
371 NodePort* port = tensor->consumer[i];
372 Node* node = port->owner;
373 int idx = port->port_index;
374
375 node->RemoveInputPort(idx);
376 }
377
378 /* remove refer from producer */
379
380 NodePort* port = tensor->producer;
381
382 if(port)
383 {
384 Node* node = port->owner;
385 int idx = port->port_index;
386
387 node->RemoveOutputPort(idx);
388 }
389
390 /* release the tensor*/
391
392 auto ir = tensor_map_.find(tensor->GetName());
393
394 if(ir != tensor_map_.end())
395 tensor_map_.erase(ir);
396
397 /* if it is my tensor */
398 if(RemoveTensorOwner(tensor))
399 delete tensor;
400
401 return true;
402}
403
404bool Graph::RemoveNode(Node* node)
405{

Callers 3

check_graph_inputsFunction · 0.80
mainFunction · 0.80
mainFunction · 0.80

Calls 5

RemoveInputPortMethod · 0.80
RemoveOutputPortMethod · 0.80
GetNameMethod · 0.80
endMethod · 0.80
sizeMethod · 0.45

Tested by 2

mainFunction · 0.64
mainFunction · 0.64