Searched refs:OutputPort (Results 1 – 14 of 14) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | graph_view.h | 92 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 …]
|
D | mutable_graph_view.h | 53 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);
|
D | mutable_graph_view.cc | 55 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 …]
|
D | graph_view.cc | 68 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()
|
D | graph_view_test.cc | 136 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()
|
D | mutable_graph_view_test.cc | 49 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/ |
D | shape_optimizer.cc | 46 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()
|
D | memory_optimizer.cc | 855 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()
|
D | function_optimizer.cc | 178 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()
|
D | loop_optimizer.cc | 744 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()
|
D | pin_to_host_optimizer.cc | 240 for (const GraphView::OutputPort& fanin : in IsNodeHostCandidate()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | latency_all_edges.cc | 83 MutableGraphView::OutputPort output_port = in OptimizeAndCollectStats()
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | topological_sort.cc | 41 GraphView::OutputPort(dep.from, Graph::kControlSlot), in MakeEphemeralEdges()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 763 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()
|