Home
last modified time | relevance | path

Searched refs:optimized_nodes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmodel_pruner.cc148 const absl::flat_hash_map<string, const NodeDef*>& optimized_nodes, in ForwardInputsInternal() argument
152 auto itr = optimized_nodes.find(node.name()); in ForwardInputsInternal()
153 if (itr != optimized_nodes.end()) { in ForwardInputsInternal()
171 optimized_nodes, graph_view); in ForwardInputsInternal()
182 absl::flat_hash_map<string, const NodeDef*>* optimized_nodes, in ForwardInputs() argument
186 /*add_as_control=*/false, new_node, *optimized_nodes, in ForwardInputs()
189 (*optimized_nodes)[new_node->name()] = new_node; in ForwardInputs()
511 absl::flat_hash_map<string, const NodeDef*> optimized_nodes; in Optimize() local
519 ForwardInputs(node, nodes_to_delete, new_node, &optimized_nodes, in Optimize()
Dmemory_optimizer.cc1322 std::unordered_set<const NodeDef*> optimized_nodes; in FindAssignNodesToRelax() local
1327 if (optimized_nodes.find(&assign_node) == optimized_nodes.end()) { in FindAssignNodesToRelax()
1329 optimized_nodes.insert(&assign_node); in FindAssignNodesToRelax()
1354 if (optimized_nodes.find(fanout_node) == optimized_nodes.end() && in FindAssignNodesToRelax()
1366 optimized_nodes.insert(assign_node_in_fanout); in FindAssignNodesToRelax()
Darithmetic_optimizer.cc385 std::vector<NodeDef*> optimized_nodes; member
396 if (!group.optimized_nodes.empty()) { in TrySimplify()
426 group->optimized_nodes.push_back(input_node); in AbsorbInputByOptimizedNodesGroup()
453 group->optimized_nodes.reserve(root_node->input_size()); in CreateOptimizedNodesGroup()
492 for (NodeDef* optimized_node : group.optimized_nodes) { in MarkAllMembersWithTag()
604 << " num_optimized_nodes=" << group.optimized_nodes.size() in RewriteOptimizedNodesGroup()
1054 << " num_optimized_nodes=" << group.optimized_nodes.size(); in RewriteOptimizedNodesGroup()
1065 auto num_nodes = /*root*/ 1 + group.optimized_nodes.size(); in RewriteOptimizedNodesGroup()
1072 std::deque<NodeDef*> optimized_nodes(group.optimized_nodes.begin(), in RewriteOptimizedNodesGroup() local
1073 group.optimized_nodes.end()); in RewriteOptimizedNodesGroup()
[all …]