HomeSort by relevance Sort by last modified time
    Searched defs:function_node (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/distributed_runtime/
cluster_function_library_runtime.cc 66 NodeDef* function_node = g->add_node(); local
67 function_node->set_name(sig.name());
68 function_node->set_op(sig.name());
71 function_node->add_input(strings::StrCat("_recv_", in.name(), "_", i));
74 function_node->set_device(target);
76 (*function_node->mutable_attr())[p.first] = p.second;
  /external/tensorflow/tensorflow/core/grappler/optimizers/data/
map_vectorization.cc 347 auto function_node = branch_func->add_node_def(); local
348 *function_node = *node;
350 function_node);
351 function_node->clear_input();
352 function_node->add_input(prev_node_output);
368 string input_arg_name = strings::StrCat(function_node->name(), "_", i);
371 function_node->add_input(input_arg_name);
378 prev_node_output = strings::StrCat(function_node->name(), ":handle:0");

Completed in 436 milliseconds