| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.h | 88 struct InputPort : public Port { struct
97 Edge(OutputPort s, InputPort d) : src(s), dst(d) {} in Edge()
109 InputPort dst;
127 InputPort GetInputPort(absl::string_view node_name, int port_id) const { in GetInputPort()
128 return InputPort(GetNode(node_name), port_id); in GetInputPort()
139 const absl::flat_hash_set<InputPort>& GetFanout( in GetFanout()
145 absl::flat_hash_set<OutputPort> GetFanin(const InputPort& port) const { in GetFanin()
170 const OutputPort GetRegularFanin(const InputPort& port) const { in GetRegularFanin()
206 absl::flat_hash_set<InputPort> GetFanouts( in GetFanouts()
208 absl::flat_hash_set<InputPort> result; in GetFanouts()
[all …]
|
| D | mutable_graph_view.cc | 111 absl::flat_hash_set<MutableGraphView::InputPort>>;
390 const absl::flat_hash_set<InputPort>& fanin_fanouts) { in UpdateMaxRegularOutputPortForRemovedFanin()
416 const absl::flat_hash_set<MutableGraphView::InputPort>&
423 const GraphView::InputPort& port) const { in GetFanin()
424 return GetFanin(MutableGraphView::InputPort(const_cast<NodeDef*>(port.node), in GetFanin()
429 const GraphView::InputPort& port) const { in GetRegularFanin()
430 return GetRegularFanin(MutableGraphView::InputPort( in GetRegularFanin()
662 absl::flat_hash_set<InputPort>* from_fanouts = &fanouts()[from_fanin]; in SwapNodeNames()
755 const InputPort& input_port) { in UpdateFanoutsInternal()
761 const InputPort& input_port) { in UpdateFanoutsInternal()
[all …]
|
| D | mutable_graph_view.h | 52 const absl::flat_hash_set<InputPort>& GetFanout(
57 const GraphView::InputPort& port) const;
60 const OutputPort GetRegularFanin(const GraphView::InputPort& port) const;
263 const absl::flat_hash_set<InputPort>& fanin_fanouts);
|
| D | graph_view.cc | 78 const auto control_port = GraphView::InputPort(node, Graph::kControlSlot); in HasControlFanin()
|
| D | graph_view_test.cc | 133 GraphView::InputPort input = graph.GetInputPort("AddN", 0); in TEST_F()
207 GraphView::InputPort input = graph.GetInputPort("d", -1); in TEST_F()
|
| D | mutable_graph_view_test.cc | 48 MutableGraphView::InputPort input_port(node, port); in CompareNodeFanins()
63 MutableGraphView::InputPort input_port(graph.GetNode(tensor_id.node()), in CompareNodeFanouts()
126 GraphView::InputPort immutable_fanout(fanout.node, fanout.port_id); in CheckGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | shape_optimizer.cc | 45 for (MutableGraphView::InputPort fanout : in Optimize()
57 graph.GetRegularFanin(MutableGraphView::InputPort(fanout.node, 1)); in Optimize()
94 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 0)); in Optimize()
96 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 1)); in Optimize()
|
| D | loop_optimizer.cc | 601 GraphView::InputPort switch_loopcond_port(&switch_node, 1); in CheckForDeadFanout()
613 GraphView::InputPort switch_input_port(&switch_node, 0); in CheckForDeadFanout()
766 SetVector<MutableGraphView::InputPort, absl::Hash<MutableGraphView::Port>> in RemoveDeadBranches()
768 for (const MutableGraphView::InputPort& port : view.GetFanout(dead)) { in RemoveDeadBranches()
782 MutableGraphView::InputPort dead = zombie_inputs.PopBack(); in RemoveDeadBranches()
802 const absl::flat_hash_set<MutableGraphView::InputPort>& index_fanout = in RemoveDeadBranches()
829 for (const MutableGraphView::InputPort& port : in RemoveDeadBranches()
840 for (const MutableGraphView::InputPort& dead_fanout : in RemoveDeadBranches()
|
| D | remapper.cc | 240 const auto input_port = GraphView::InputPort(bias_add, 0); in FindConv2DWithBias()
271 const auto input_port = GraphView::InputPort(relu, 0); in FindConv2DWithBiasAndRelu()
303 const auto bias_input_port = GraphView::InputPort(bias_add, 0); in FindConv2DWithSqueezeAndBias()
321 const auto squeeze_input_port = GraphView::InputPort(squeeze.node, 0); in FindConv2DWithSqueezeAndBias()
368 const auto input_port = GraphView::InputPort(batch_norm, 0); in FindConv2DWithBatchNorm()
400 const auto input_port = GraphView::InputPort(node, 0); in FindConv2DWithBatchNormAndRelu()
|
| D | memory_optimizer.cc | 881 MutableGraphView::InputPort input; in IsSwappable()
897 MutableGraphView::InputPort swap; in FindSwapOutTrigger()
905 const absl::flat_hash_set<MutableGraphView::InputPort>& fanout = in FindSwapOutTrigger()
924 static bool IsSwappable(MutableGraphView::InputPort input) { in IsSwappable()
943 std::vector<MutableGraphView::InputPort> uses_left;
1041 for (MutableGraphView::InputPort input : graph.GetFanout(port)) { in IdentifySwappingCandidates()
1094 for (const MutableGraphView::InputPort fanout_to_swap : in IdentifySwappingCandidates()
|
| D | function_optimizer.cc | 177 using InputPort = int; typedef
185 absl::flat_hash_map<InputPort, string> const_inputs;
237 using ConstInput = std::pair<const InputPort, string>; in AbslHashValue()
239 hashes.push_back(hash<InputPort>()(const_input.first)); in AbslHashValue()
1601 GraphView::InputPort control_input_port = in InlineIndirectFunctionCall()
1621 [](const GraphView::InputPort port) { return port.node->name(); }); in InlineIndirectFunctionCall()
|
| D | pin_to_host_optimizer.cc | 361 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | frame.cc | 52 absl::flat_hash_set<GraphView::InputPort> fanouts = in InferFromGraphView()
55 for (const GraphView::InputPort& fanout : fanouts) { in InferFromGraphView()
|
| D | topological_sort.cc | 42 GraphView::InputPort(dep.to, Graph::kControlSlot)); in MakeEphemeralEdges()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.cc | 231 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), 0); in GetInputNode()
238 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), i); in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 762 const GraphView::InputPort port(node, dst_input); in UpdateNode()
1844 GraphView::InputPort port(node, 0); in UpdateEnter()
2016 GraphView::InputPort inp(enqueue_node, i); in UpdateEnqueue()
2064 for (const GraphView::InputPort& fanout : in InferStatically()
2068 for (const GraphView::InputPort& fanout : in InferStatically()
2192 GraphView::InputPort input(&node, -1); in InferStatically()
|