MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / SetNewNodesFanins

Method SetNewNodesFanins

tensorflow/core/grappler/utils/graph_view.cc:1255–1272  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1253}
1254
1255void MutableGraphView::SetNewNodesFanins(
1256 const std::vector<int>& new_node_indices) {
1257 auto new_node = mutation_.new_nodes_.begin();
1258 for (const int new_node_index : new_node_indices) {
1259 MutableNodeView& new_node_view = nodes_[new_node_index];
1260 NodeDef* new_node_def = new_node_view.node();
1261 new_node_def->mutable_input()->Reserve(new_node->num_regular_fanins +
1262 new_node->controlling_fanins.size());
1263 for (const SafeTensorId& fanin : new_node->regular_fanins) {
1264 AddRegularFaninInternal(&new_node_view, fanin);
1265 new_node_def->add_input(SafeTensorIdToString(fanin));
1266 }
1267 for (const string& control_to_add : new_node->controlling_fanins) {
1268 AddControllingFaninInternal(&new_node_view, control_to_add);
1269 }
1270 ++new_node;
1271 }
1272}
1273
1274inline void MutableGraphView::RemoveAllFaninFanoutInternal(
1275 MutableNodeView* node_view) {

Callers

nothing calls this directly

Calls 7

SafeTensorIdToStringFunction · 0.85
beginMethod · 0.45
nodeMethod · 0.45
ReserveMethod · 0.45
mutable_inputMethod · 0.45
sizeMethod · 0.45
add_inputMethod · 0.45

Tested by

no test coverage detected