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.cc130 const absl::flat_hash_map<string, const NodeDef*>& optimized_nodes, in ForwardInputsInternal() argument
134 auto itr = optimized_nodes.find(node.name()); in ForwardInputsInternal()
135 if (itr != optimized_nodes.end()) { in ForwardInputsInternal()
153 optimized_nodes, graph_view); in ForwardInputsInternal()
164 absl::flat_hash_map<string, const NodeDef*>* optimized_nodes, in ForwardInputs() argument
168 /*add_as_control=*/false, new_node, *optimized_nodes, in ForwardInputs()
171 (*optimized_nodes)[new_node->name()] = new_node; in ForwardInputs()
499 absl::flat_hash_map<string, const NodeDef*> optimized_nodes; in Optimize() local
506 ForwardInputs(node, nodes_to_delete, new_node, &optimized_nodes, in Optimize()
Dmemory_optimizer.cc1276 std::unordered_set<const NodeDef*> optimized_nodes; in RelaxAllocatorConstraints() local
1281 if (optimized_nodes.find(&assign_node) == optimized_nodes.end()) { in RelaxAllocatorConstraints()
1283 optimized_nodes.insert(&assign_node); in RelaxAllocatorConstraints()
1308 if (optimized_nodes.find(fanout_node) == optimized_nodes.end() && in RelaxAllocatorConstraints()
1320 optimized_nodes.insert(assign_node_in_fanout); in RelaxAllocatorConstraints()
Darithmetic_optimizer.cc298 std::vector<NodeDef*> optimized_nodes; member
309 if (!group.optimized_nodes.empty()) { in TrySimplify()
339 group->optimized_nodes.push_back(input_node); in AbsorbInputByOptimizedNodesGroup()
366 group->optimized_nodes.reserve(root_node->input_size()); in CreateOptimizedNodesGroup()
405 for (NodeDef* optimized_node : group.optimized_nodes) { in MarkAllMembersWithTag()
517 << " num_optimized_nodes=" << group.optimized_nodes.size() in RewriteOptimizedNodesGroup()
964 << " num_optimized_nodes=" << group.optimized_nodes.size(); in RewriteOptimizedNodesGroup()
975 auto num_nodes = /*root*/ 1 + group.optimized_nodes.size(); in RewriteOptimizedNodesGroup()
982 std::deque<NodeDef*> optimized_nodes(group.optimized_nodes.begin(), in RewriteOptimizedNodesGroup() local
983 group.optimized_nodes.end()); in RewriteOptimizedNodesGroup()
[all …]