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()
111 absl::flat_hash_map<MutableGraphView::OutputPort,
134 MutableGraphView::OutputPort port; in SwapRegularFanoutInputs()
175 const MutableGraphView::OutputPort& from_port, in SwapFanoutsMapValues()
177 const MutableGraphView::OutputPort& to_port, in SwapFanoutsMapValues()
204 MutableGraphView::OutputPort from_port(from, i); in SwapRegularFanoutsAndMaxPortValues()
207 MutableGraphView::OutputPort to_port(to, i); in SwapRegularFanoutsAndMaxPortValues()
219 MutableGraphView::OutputPort from_port(from_node, i); in SwapRegularFanoutsAndMaxPortValues()
221 MutableGraphView::OutputPort to_port(to_node, i); in SwapRegularFanoutsAndMaxPortValues()
331 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.cc137 GraphView::OutputPort fanin = graph.GetRegularFanin(input); in TEST_F()
148 GraphView::OutputPort output = graph.GetOutputPort("AddN", 0); in TEST_F()
192 GraphView::OutputPort output = graph.GetOutputPort("a", -1); in TEST_F()
286 EXPECT_EQ(d_output_0, GraphView::OutputPort(b_node, 0)); in TEST_F()
288 EXPECT_EQ(d_output_1, GraphView::OutputPort(c_node, 0)); in TEST_F()
290 EXPECT_EQ(d_output_2, GraphView::OutputPort()); in TEST_F()
292 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.cc80 graph.GetFanout(MutableGraphView::OutputPort(&node, 0))) { in Optimize()
90 const MutableGraphView::OutputPort reduce_indices = in Optimize()
148 const MutableGraphView::OutputPort input1 = in Optimize()
150 const MutableGraphView::OutputPort input2 = in Optimize()
Dmemory_optimizer.cc886 MutableGraphView::OutputPort output) { in IsSwappable()
915 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); in IsSwappable()
931 MutableGraphView::OutputPort generator = view.GetRegularFanin(swap); in FindSwapOutTrigger()
972 MutableGraphView::OutputPort port;
1065 MutableGraphView::OutputPort port = in IdentifySwappingCandidates()
Dloop_optimizer.cc769 absl::flat_hash_set<GraphView::OutputPort> identity_switches; in RemoveDeadBranches()
791 GraphView::OutputPort dead(&node, dead_fanout); in RemoveDeadBranches()
829 MutableGraphView::OutputPort value_index(dead.node, 1); in RemoveDeadBranches()
Dauto_mixed_precision.cc994 NodeDef BuildCastNode(const MutableGraphView::OutputPort& src, bool to_f16,
1019 const MutableGraphView::OutputPort& src, bool to_f16, in BuildCastNode()
1731 GraphView::OutputPort output_port(&node, 0); in ForceColorMatchOnRecurrentEdges()
1833 MutableGraphView::OutputPort prev_output = in NodeImplicitlyReadsNonResourceVariable()
1858 MutableGraphView::OutputPort src(node, 0); in MakeCastsAllowIfAllOutputsAllow()
1919 MutableGraphView::OutputPort src(node, output_port); in ChangeTypeAttrsAndAddCasts()
Dpin_to_host_optimizer.cc244 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate()
Dfunction_optimizer.cc161 using OutputPort = int; typedef
165 absl::flat_hash_set<OutputPort> active_outputs;
204 hash<OutputPort>()); in AbslHashValue()
/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.cc976 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); in UpdateNode()
2278 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); in UpdateEnter()
2450 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); in UpdateEnqueue()
2637 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); in InferStatically()