Searched refs:feed_nodes (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 587 const absl::flat_hash_set<string>& feed_nodes) { in IsReallyConstant() argument
592 return feed_nodes.find(node.name()) == feed_nodes.end(); in IsReallyConstant()
597 const absl::flat_hash_set<string>& feed_nodes, in CheckForDeadFanout() argument
606 if (IsReallyConstant(*switch_predicate, feed_nodes)) { in CheckForDeadFanout()
645 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
661 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
669 if (IsReallyConstant(*node, feed_nodes)) { in CheckForDeadFanout()
725 absl::flat_hash_set<string> feed_nodes; in Optimize() local
727 feed_nodes.insert(NodeName(feed.first)); in Optimize()
730 feed_nodes, optimized_graph)); in Optimize()
[all …]
|
| D | graph_optimizer_stage.h | 52 gtl::FlatSet<string>* feed_nodes, in GraphOptimizerContext()
58 feed_nodes(feed_nodes), in GraphOptimizerContext()
65 gtl::FlatSet<string>* feed_nodes; member
|
| D | loop_optimizer.h | 64 const absl::flat_hash_set<string>& feed_nodes,
|
| D | function_optimizer.cc | 394 absl::flat_hash_set<absl::string_view> feed_nodes; in InferTrulyConstNodes() local
396 feed_nodes.insert(feed.first); in InferTrulyConstNodes()
401 if (IsConstant(node) && !feed_nodes.contains(node.name())) { in InferTrulyConstNodes()
|
| D | arithmetic_optimizer.cc | 3088 return ctx().feed_nodes->find(node.name()) == ctx().feed_nodes->end(); in IsReallyConstant()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 309 std::set<string> feed_nodes; in Init() local
311 auto iter_and_inserted_flag = feed_nodes.insert(f.first); in Init()
425 feed_nodes.find(curr_node->name()) != feed_nodes.end(); in Init()
438 feed_nodes.erase(curr_node->name()); in Init()
454 if (!feed_nodes.empty()) { in Init()
459 << str_util::Join(feed_nodes, ","); in Init()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | grappler_item_builder.h | 51 std::set<string> feed_nodes; member
|
| D | grappler_item_builder.cc | 201 for (const auto& feed_node : cfg.feed_nodes) { in GrapplerItemFromMetaGraphDef()
499 if (cfg.feed_nodes.empty()) { in GrapplerItemFromMetaGraphDef()
504 } else if (cfg.feed_nodes.count(node.name()) > 0) { in GrapplerItemFromMetaGraphDef()
|
| D | grappler_item_builder_test.cc | 326 config.feed_nodes.insert("x"); in TEST_F()
|