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