Searched refs:feed_nodes (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | loop_optimizer.cc | 589 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 …]
|
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 | 68 const absl::flat_hash_set<string>& feed_nodes,
|
D | function_optimizer.cc | 342 absl::flat_hash_set<absl::string_view> feed_nodes; in InferTrulyConstNodes() local 344 feed_nodes.insert(feed.first); in InferTrulyConstNodes() 349 if (IsConstant(node) && !feed_nodes.contains(node.name())) { in InferTrulyConstNodes() 1227 NodeNames feed_nodes; in InlineFunctionCalls() local 1228 feed_nodes.reserve(item.feed.size()); in InlineFunctionCalls() 1230 feed_nodes.insert(ParseTensorName(feed.first).node()); in InlineFunctionCalls() 1269 if (feed_nodes.contains(n->name())) continue; in InlineFunctionCalls()
|
D | arithmetic_optimizer.cc | 250 return ctx().feed_nodes->find(node.name()) == ctx().feed_nodes->end(); in IsReallyConstant()
|
/external/tensorflow/tensorflow/core/grappler/ |
D | grappler_item_builder.h | 50 std::set<string> feed_nodes; member
|
D | grappler_item_builder.cc | 178 if (cfg.feed_nodes.empty()) { in UpdatePlaceholderShape() 183 } else if (cfg.feed_nodes.count(node->name()) > 0) { in UpdatePlaceholderShape() 301 for (const auto& feed_node : cfg.feed_nodes) { in GrapplerItemFromMetaGraphDef()
|
D | grappler_item_builder_test.cc | 326 config.feed_nodes.insert("x"); in TEST_F()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler.cc | 433 std::set<string> feed_nodes; in Init() local 436 auto iter_and_inserted_flag = feed_nodes.insert(f.first); in Init() 547 feed_nodes.find(curr_node->name()) != feed_nodes.end(); in Init() 559 feed_nodes.erase(curr_node->name()); in Init() 576 if (!feed_nodes.empty()) { in Init() 581 << absl::StrJoin(feed_nodes, ","); in Init()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | graph_execution_state.cc | 695 absl::flat_hash_set<absl::string_view> feed_nodes; in OptimizeGraph() 707 feed_nodes.insert(feed.node()); in OptimizeGraph() 717 if (feed_nodes.find(node->name()) == feed_nodes.end()) continue; in OptimizeGraph()
|