Home
last modified time | relevance | path

Searched refs:feed_nodes (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc589 const absl::flat_hash_set<string>& feed_nodes) { in IsReallyConstant() argument
594 return feed_nodes.find(node.name()) == feed_nodes.end(); in IsReallyConstant()
599 const absl::flat_hash_set<string>& feed_nodes, in CheckForDeadFanout() argument
608 if (IsReallyConstant(*switch_predicate, feed_nodes)) { in CheckForDeadFanout()
652 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
669 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
677 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
750 absl::flat_hash_set<string> feed_nodes; in Optimize() local
752 feed_nodes.insert(NodeName(feed.first)); in Optimize()
755 feed_nodes, optimized_graph)); in Optimize()
[all …]
Dgraph_optimizer_stage.h52 gtl::FlatSet<string>* feed_nodes, in GraphOptimizerContext()
58 feed_nodes(feed_nodes), in GraphOptimizerContext()
65 gtl::FlatSet<string>* feed_nodes; member
Dloop_optimizer.h65 const absl::flat_hash_set<string>& feed_nodes,
Dfunction_optimizer.cc345 absl::flat_hash_set<absl::string_view> feed_nodes; in InferTrulyConstNodes() local
347 feed_nodes.insert(feed.first); in InferTrulyConstNodes()
352 if (IsConstant(node) && !feed_nodes.contains(node.name())) { in InferTrulyConstNodes()
1237 NodeNames feed_nodes; in InlineFunctionCalls() local
1238 feed_nodes.reserve(item.feed.size()); in InlineFunctionCalls()
1240 feed_nodes.insert(ParseTensorName(feed.first).node()); in InlineFunctionCalls()
1279 if (feed_nodes.contains(n->name())) continue; in InlineFunctionCalls()
Darithmetic_optimizer.cc312 return ctx().feed_nodes->find(node.name()) == ctx().feed_nodes->end(); in IsReallyConstant()
/external/tensorflow/tensorflow/core/grappler/
Dgrappler_item_builder.h50 std::set<string> feed_nodes; member
Dgrappler_item_builder.cc178 if (cfg.feed_nodes.empty()) { in UpdatePlaceholderShape()
183 } else if (cfg.feed_nodes.count(node->name()) > 0) { in UpdatePlaceholderShape()
302 for (const auto& feed_node : cfg.feed_nodes) { in GrapplerItemFromMetaGraphDef()
Dgrappler_item_builder_test.cc326 config.feed_nodes.insert("x"); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler.cc438 std::set<string> feed_nodes; in Init() local
441 auto iter_and_inserted_flag = feed_nodes.insert(f.first); in Init()
552 feed_nodes.find(curr_node->name()) != feed_nodes.end(); in Init()
564 feed_nodes.erase(curr_node->name()); in Init()
581 if (!feed_nodes.empty()) { in Init()
586 << absl::StrJoin(feed_nodes, ","); in Init()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc700 absl::flat_hash_set<absl::string_view> feed_nodes; in OptimizeGraph()
712 feed_nodes.insert(feed.node()); in OptimizeGraph()
722 if (feed_nodes.find(node->name()) == feed_nodes.end()) continue; in OptimizeGraph()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtrt_convert_api.cc72 item_config.feed_nodes.insert(name); in RunGrappler()