| /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.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 | mutable_graph_view.cc | 112 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()
758 const InputPort& input_port) { in UpdateFanoutsInternal()
764 const InputPort& input_port) { in UpdateFanoutsInternal()
[all …]
|
| D | graph_view.cc | 78 const auto control_port = GraphView::InputPort(node, Graph::kControlSlot); in HasControlFanin()
|
| D | graph_view_test.cc | 134 GraphView::InputPort input = graph.GetInputPort("AddN", 0); in TEST_F()
208 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 | 79 for (MutableGraphView::InputPort fanout : in Optimize()
91 graph.GetRegularFanin(MutableGraphView::InputPort(fanout.node, 1)); in Optimize()
149 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 0)); in Optimize()
151 graph.GetRegularFanin(MutableGraphView::InputPort(&node, 1)); in Optimize()
|
| D | loop_optimizer.cc | 603 GraphView::InputPort switch_loopcond_port(&switch_node, 1); in CheckForDeadFanout()
619 GraphView::InputPort switch_input_port(&switch_node, 0); in CheckForDeadFanout()
794 SetVector<MutableGraphView::InputPort, absl::Hash<MutableGraphView::Port>> in RemoveDeadBranches()
796 for (const MutableGraphView::InputPort& port : view.GetFanout(dead)) { in RemoveDeadBranches()
810 MutableGraphView::InputPort dead = zombie_inputs.PopBack(); in RemoveDeadBranches()
830 const absl::flat_hash_set<MutableGraphView::InputPort>& index_fanout = in RemoveDeadBranches()
857 for (const MutableGraphView::InputPort& port : in RemoveDeadBranches()
868 for (const MutableGraphView::InputPort& dead_fanout : in RemoveDeadBranches()
|
| D | memory_optimizer.cc | 912 MutableGraphView::InputPort input; in IsSwappable()
928 MutableGraphView::InputPort swap; in FindSwapOutTrigger()
936 const absl::flat_hash_set<MutableGraphView::InputPort>& fanout = in FindSwapOutTrigger()
955 static bool IsSwappable(MutableGraphView::InputPort input) { in IsSwappable()
974 std::vector<MutableGraphView::InputPort> uses_left;
1076 for (MutableGraphView::InputPort input : graph.GetFanout(port)) { in IdentifySwappingCandidates()
1129 for (const MutableGraphView::InputPort fanout_to_swap : in IdentifySwappingCandidates()
|
| D | function_optimizer.cc | 160 using InputPort = int; typedef
168 absl::flat_hash_map<InputPort, string> const_inputs;
220 using ConstInput = std::pair<const InputPort, string>; in AbslHashValue()
222 hashes.push_back(hash<InputPort>()(const_input.first)); in AbslHashValue()
|
| D | pin_to_host_optimizer.cc | 357 for (const GraphView::InputPort& fanout : graph.GetFanouts(*node, false)) { in Optimize()
|
| D | auto_mixed_precision.cc | 1832 GraphView::InputPort node_input(&node, 0); in NodeImplicitlyReadsNonResourceVariable()
1860 for (const MutableGraphView::InputPort& dst : fanout) { in MakeCastsAllowIfAllOutputsAllow()
1923 for (const MutableGraphView::InputPort& dst : fanout) { in ChangeTypeAttrsAndAddCasts()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | topological_sort.cc | 42 GraphView::InputPort(dep.to, Graph::kControlSlot)); in MakeEphemeralEdges()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | compute_batch_size_op.cc | 77 grappler::GraphView::InputPort input_port = in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils.cc | 273 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), 0); in GetInputNode()
280 MutableGraphView::InputPort input_port = graph.GetInputPort(node.name(), i); in GetInputNode()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 975 const GraphView::InputPort port(node, dst_input); in UpdateNode()
2277 GraphView::InputPort port(node, 0); in UpdateEnter()
2449 GraphView::InputPort inp(enqueue_node, i); in UpdateEnqueue()
2499 for (const GraphView::InputPort& fanout : in InferStatically()
2503 for (const GraphView::InputPort& fanout : in InferStatically()
2632 GraphView::InputPort input(&node, -1); in InferStatically()
|