Searched refs:from_fanin (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 661 MutableGraphView::OutputPort from_fanin(node, idx); in SwapNodeNames() local
662 absl::flat_hash_set<InputPort>* from_fanouts = &fanouts()[from_fanin]; in SwapNodeNames()
664 UpdateMaxRegularOutputPortForRemovedFanin(from_fanin, *from_fanouts); in SwapNodeNames()
1233 const TensorId& from_fanin, in UpdateFanin() argument
1235 auto error_status = [node_name, from_fanin, to_fanin](absl::string_view msg) { in UpdateFanin()
1238 node_name, from_fanin.ToString(), to_fanin.ToString()); in UpdateFanin()
1242 TF_RETURN_IF_ERROR(CheckFaninIsValid(from_fanin, error_status)); in UpdateFanin()
1246 NodeDef* from_fanin_node = GetNode(from_fanin.node()); in UpdateFanin()
1248 CheckNodeExists(from_fanin.node(), from_fanin_node, error_status)); in UpdateFanin()
1263 if (node_name == from_fanin.node() || node_name == to_fanin.node()) { in UpdateFanin()
[all …]
|
| D | mutable_graph_view.h | 219 Status UpdateFanin(absl::string_view node_name, const TensorId& from_fanin,
|
| D | mutable_graph_view_test.cc | 1538 const TensorId& from_fanin, const TensorId& to_fanin, in TestUpdateFanin() argument
1555 Status s = graph.UpdateFanin(node_name, from_fanin, to_fanin); in TestUpdateFanin()
1563 CheckFanoutRemoved(graph, from_fanin, node_name); in TestUpdateFanin()
|