Searched refs:fanin_port (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.cc | 397 OutputPort fanin_port(fanin.node, i); in UpdateMaxRegularOutputPortForRemovedFanin() local
398 if (!fanouts()[fanin_port].empty()) { in UpdateMaxRegularOutputPortForRemovedFanin()
937 OutputPort fanin_port(nodes()[tensor_id.node()], tensor_id.index()); in AddRegularFaninByPort() local
938 absl::flat_hash_set<InputPort>* fanouts_set = &fanouts()[fanin_port]; in AddRegularFaninByPort()
944 OutputPort fanin_port(fanin_node, fanin.index()); in AddRegularFaninByPort() local
945 fanouts()[fanin_port].insert({node, port}); in AddRegularFaninByPort()
946 UpdateMaxRegularOutputPortForAddedFanin(fanin_port); in AddRegularFaninByPort()
1030 OutputPort fanin_port(fanin_node, fanin.index()); in AddControllingFanin() local
1039 control_node = GetOrCreateIdentityConsumingSwitch(fanin_port); in AddControllingFanin()
1048 auto remove_input = [this, node](const OutputPort& fanin_port, in RemoveRegularFaninInternal()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 202 absl::string_view fanin_name, int fanin_port) { in VerifyRegularFaninMatch() argument
206 EXPECT_EQ(fanin.index(), fanin_port); in VerifyRegularFaninMatch()
|
| D | generic_layout_optimizer_transposer.cc | 381 auto& fanin_port = dst_node->GetRegularFanin(dst_port); in UpdateFaninEdgesWithOp() local
382 auto* fanin_node_view = fanin_port.node_view(); in UpdateFaninEdgesWithOp()
389 /*is_src_format_to_dst_format=*/true, fanin_port.index(), in UpdateFaninEdgesWithOp()
|
| D | generic_layout_optimizer_transposer_test.cc | 75 absl::string_view fanin_name, int fanin_port) { in VerifyRegularFaninMatch() argument
79 EXPECT_EQ(fanin.index(), fanin_port); in VerifyRegularFaninMatch()
|