HomeSort by relevance Sort by last modified time
    Searched refs:feed_nodes (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/grappler/
grappler_item_builder.h 53 std::set<string> feed_nodes; member in struct:tensorflow::grappler::ItemConfig
grappler_item_builder.cc 169 for (const auto& feed_node : cfg.feed_nodes) {
371 if (cfg.feed_nodes.empty()) {
374 } else if (cfg.feed_nodes.count(node.name()) > 0) {
  /external/tensorflow/tensorflow/core/grappler/costs/
virtual_scheduler.cc 319 std::set<string> feed_nodes; local
321 auto iter_and_inserted_flag = feed_nodes.insert(f.first);
426 feed_nodes.find(curr_node->name()) != feed_nodes.end();
437 feed_nodes.erase(curr_node->name());
453 if (!feed_nodes.empty()) {
456 str_util::Join(feed_nodes, ",")));
    [all...]

Completed in 316 milliseconds