| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view.h | 136 // Adds regular fanin `fanin` to node `node_name`. If the node or fanin do not
137 // exist in the graph, nothing will be modified in the graph. Otherwise fanin
141 Status AddRegularFanin(absl::string_view node_name, const TensorId& fanin);
143 // Adds regular fanin `fanin` to node `node_name` at port `port`. If the node
144 // or fanin do not exist in the graph, nothing will be modified in the graph.
145 // Otherwise fanin will be inserted at port `port`. Control dependencies will
152 const TensorId& fanin);
[all...] |
| mutable_graph_view.cc | 266 Status CheckFaninIsRegular(const TensorId& fanin, ErrorHandler handler) {
267 if (!IsTensorIdRegular(fanin)) {
268 return handler(absl::Substitute("fanin '$0' must be a regular tensor id",
269 fanin.ToString()));
274 Status CheckFaninIsValid(const TensorId& fanin, ErrorHandler handler) {
275 if (!IsTensorIdPortValid(fanin)) {
276 return handler(absl::Substitute("fanin '$0' must be a valid tensor id",
277 fanin.ToString()));
283 const TensorId& fanin, ErrorHandler handler) {
284 if (node_name == fanin.node())
[all...] |
| graph_view_test.cc | 136 GraphView::OutputPort fanin = graph.GetRegularFanin(input); local
137 EXPECT_EQ(fanin.node->name(), "Square");
138 EXPECT_EQ(fanin.port_id, 0);
143 fanin = graph.GetRegularFanin(input);
144 EXPECT_EQ(fanin.node->name(), "Square_1");
145 EXPECT_EQ(fanin.port_id, 0);
210 auto fanin = graph.GetFanin(input); local
211 EXPECT_EQ(fanin.size(), 1);
212 EXPECT_EQ((*fanin.begin()).node->name(), "a");
213 EXPECT_EQ((*fanin.begin()).port_id, -1)
[all...] |
| graph_view.h | 144 // Gets the output port(s) in the immediate fanin of an input port.
154 // Collect fanin for the control input.
168 // fanin. If port.port_id is out of range or is a control dependency, then an
181 // Checks if a tensor id is a fanin of the node.
182 bool HasFanin(const NodeDefT& node, const TensorId& fanin) const {
184 if (end == 0 || fanin.index() < -1) {
191 if (fanin.index() > -1) {
197 if (ParseTensorName(node.input(i)) == fanin) {
226 // Gets all the output ports in the immediate fanin of a node. Include the
244 // Gets the number of ports in the immediate fanin of a node. Count th
[all...] |
| grappler_item.cc | 70 std::vector<const NodeDef*> fanin = ComputeTransitiveFanin(graph, init_ops); local
72 for (const NodeDef* node : fanin) {
|
| mutable_graph_view_test.cc | 115 for (const auto& fanin : mutable_fanins) {
116 GraphView::OutputPort immutable_fanin(fanin.node, fanin.port_id);
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| static_schedule.cc | 88 const NodeDef* fanin = it->second; local
89 fanouts[fanin].push_back(&node);
148 const NodeDef* fanin = it->second; local
149 pending_fanouts[fanin] += 1;
176 const NodeDef* fanin = name_map[NodeName(fanin_name)]; local
177 (*required_times)[fanin] =
178 std::min((*required_times)[fanin], required_time);
180 int pending = pending_fanouts[fanin];
185 ready_nodes.push_back(fanin);
187 pending_fanouts[fanin]--
[all...] |
| pin_to_host_optimizer.cc | 121 for (const auto& fanin : graph.GetFanins(node, false)) {
124 graph, properties, *fanin.node, fanin.port_id, &fanin_candidate));
240 for (const GraphView::OutputPort& fanin :
244 graph, properties, *fanin.node, fanin.port_id, &fanin_candidate));
|
| memory_optimizer.cc | 592 // Compute a topological ordering for the node fanin.
884 MutableGraphView::OutputPort fanin = graph.GetRegularFanin(input); local
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| controller.py | 42 for fanin in self._get_node_fanin(node):
43 self._fanout[fanin.name].append(node)
67 # Update the fanout of the fanin to bypass the node
68 for fanin in self._get_node_fanin(node):
69 fanout_of_fanin = self.get_node_fanout(fanin)
132 # Get the nodes in the immediate fanin of node.
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| traversal.cc | 106 // Now enqueue the fanin/fanout nodes.
108 for (const int fanin : graph_view.GetFanin(w.node)) {
109 stack.emplace_back(fanin, false, w.node);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| auto_shard.cc | 167 for (const auto& fanin : graph->GetFanins(node, true)) {
169 RemoveShuffleDataset(graph, *fanin.node, nodes_to_delete));
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties.cc | 763 const GraphView::OutputPort fanin = graph_.GetRegularFanin(port); local
764 int src_output = fanin.port_id;
765 const NodeDef* src = fanin.node;
1845 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(port); local
2017 GraphView::OutputPort fanin = shape_refiner->graph().GetRegularFanin(inp); local
2197 GraphView::OutputPort fanin = graph_view.GetRegularFanin(input); local
[all...] |
