Home
last modified time | relevance | path

Searched refs:OutputPort (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view.h92 struct OutputPort : public Port { struct
97 Edge(OutputPort s, InputPort d) : src(s), dst(d) {} in Edge()
108 OutputPort src;
134 OutputPort GetOutputPort(absl::string_view node_name, int port_id) const { in GetOutputPort()
135 return OutputPort(GetNode(node_name), port_id); in GetOutputPort()
140 const OutputPort& port) const { in GetFanout()
145 absl::flat_hash_set<OutputPort> GetFanin(const InputPort& port) const { in GetFanin()
147 OutputPort regular_fanin = GetRegularFanin(port); in GetFanin()
155 absl::flat_hash_set<OutputPort> result; in GetFanin()
170 const OutputPort GetRegularFanin(const InputPort& port) const { in GetRegularFanin()
[all …]
Dmutable_graph_view.h53 const GraphView::OutputPort& port) const;
56 absl::flat_hash_set<OutputPort> GetFanin(
60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
262 const OutputPort& fanin,
267 void UpdateMaxRegularOutputPortForAddedFanin(const OutputPort& fanin);
289 bool AddFaninInternal(NodeDef* node, const OutputPort& fanin);
299 const OutputPort& fanin, string* error_msg);
304 NodeDef* GetOrCreateIdentityConsumingSwitch(const OutputPort& fanin);
307 bool RemoveRegularFaninInternal(NodeDef* node, const OutputPort& fanin);
Dmutable_graph_view.cc55 bool IsOutputPortControlling(const MutableGraphView::OutputPort& port) { in IsOutputPortControlling()
110 absl::flat_hash_map<MutableGraphView::OutputPort,
133 MutableGraphView::OutputPort port; in SwapRegularFanoutInputs()
174 const MutableGraphView::OutputPort& from_port, in SwapFanoutsMapValues()
176 const MutableGraphView::OutputPort& to_port, in SwapFanoutsMapValues()
203 MutableGraphView::OutputPort from_port(from, i); in SwapRegularFanoutsAndMaxPortValues()
206 MutableGraphView::OutputPort to_port(to, i); in SwapRegularFanoutsAndMaxPortValues()
218 MutableGraphView::OutputPort from_port(from_node, i); in SwapRegularFanoutsAndMaxPortValues()
220 MutableGraphView::OutputPort to_port(to_node, i); in SwapRegularFanoutsAndMaxPortValues()
330 const MutableGraphView::OutputPort& fanin) { in GeneratedNameForIdentityConsumingSwitch()
[all …]
Dgraph_view.cc68 const auto output = GraphView::OutputPort(node, port); in HasSingleFanoutNode()
73 const auto output = GraphView::OutputPort(node, port); in HasFanouts()
83 const auto control_port = GraphView::OutputPort(node, Graph::kControlSlot); in HasControlFanout()
Dgraph_view_test.cc136 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F()
147 GraphView::OutputPort output = graph.GetOutputPort("AddN", 0); in TEST_F()
191 GraphView::OutputPort output = graph.GetOutputPort("a", -1); in TEST_F()
285 EXPECT_EQ(d_output_0, GraphView::OutputPort(b_node, 0)); in TEST_F()
287 EXPECT_EQ(d_output_1, GraphView::OutputPort(c_node, 0)); in TEST_F()
289 EXPECT_EQ(d_output_2, GraphView::OutputPort()); in TEST_F()
291 EXPECT_EQ(d_output_control, GraphView::OutputPort()); in TEST_F()
Dmutable_graph_view_test.cc49 MutableGraphView::OutputPort output_port = in CompareNodeFanins()
116 GraphView::OutputPort immutable_fanin(fanin.node, fanin.port_id); in CheckGraph()
1186 MutableGraphView::OutputPort output_port = in CheckFanoutRemoved()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc46 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
56 const MutableGraphView::OutputPort reduce_indices = in Optimize()
93 const MutableGraphView::OutputPort input1 = in Optimize()
95 const MutableGraphView::OutputPort input2 = in Optimize()
Dmemory_optimizer.cc855 MutableGraphView::OutputPort output) { in IsSwappable()
884 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable()
900 MutableGraphView::OutputPort generator = view.GetRegularFanin(swap); in FindSwapOutTrigger()
941 MutableGraphView::OutputPort port;
1030 MutableGraphView::OutputPort port = in IdentifySwappingCandidates()
Dfunction_optimizer.cc178 using OutputPort = int; typedef
182 absl::flat_hash_set<OutputPort> active_outputs;
221 hash<OutputPort>()); in AbslHashValue()
1510 const GraphView::OutputPort output_port = in PlaceInlinedFunctionBody()
1603 GraphView::OutputPort control_output_port = in InlineIndirectFunctionCall()
1611 [](const GraphView::OutputPort port) { return port.node->name(); }); in InlineIndirectFunctionCall()
Dloop_optimizer.cc744 absl::flat_hash_set<GraphView::OutputPort> identity_switches; in RemoveDeadBranches()
763 GraphView::OutputPort dead(&node, dead_fanout); in RemoveDeadBranches()
801 MutableGraphView::OutputPort value_index(dead.node, 1); in RemoveDeadBranches()
Dpin_to_host_optimizer.cc240 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dlatency_all_edges.cc83 MutableGraphView::OutputPort output_port = in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/utils/
Dtopological_sort.cc41 GraphView::OutputPort(dep.from, Graph::kControlSlot), in MakeEphemeralEdges()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc763 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
1845 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2017 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
2197 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically()