Searched refs:function_node (Results 1 – 5 of 5) sorted by relevance
66 NodeDef* function_node = g->add_node(); in ConstructFunctionGraph() local67 function_node->set_name(sig.name()); in ConstructFunctionGraph()68 function_node->set_op(sig.name()); in ConstructFunctionGraph()71 function_node->add_input(strings::StrCat("_recv_", in.name(), "_", i)); in ConstructFunctionGraph()74 function_node->set_device(target); in ConstructFunctionGraph()76 (*function_node->mutable_attr())[p.first] = p.second; in ConstructFunctionGraph()
129 for (NodeDef& function_node : *nodes_to_fuse) { in RenameFunctionNodes()131 gtl::FindOrNull(changed_node_names, function_node.name())) { in RenameFunctionNodes()132 function_node.set_name(*new_name); in RenameFunctionNodes()135 for (string& input : *function_node.mutable_input()) { in RenameFunctionNodes()194 for (NodeDef& function_node : *nodes_to_fuse) { in GetUniqueSignature()195 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()215 for (NodeDef& function_node : *nodes_to_fuse) { in FuseFunctionNodes()216 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()
347 auto function_node = branch_func->add_node_def(); in AddBranch() local348 *function_node = *node; in AddBranch()350 function_node); in AddBranch()351 function_node->clear_input(); in AddBranch()352 function_node->add_input(prev_node_output); in AddBranch()368 string input_arg_name = strings::StrCat(function_node->name(), "_", i); in AddBranch()371 function_node->add_input(input_arg_name); in AddBranch()378 prev_node_output = strings::StrCat(function_node->name(), ":handle:0"); in AddBranch()
115 for (const NodeDef& function_node : function.node_def()) { in TEST() local116 EXPECT_NE(node.name(), function_node.name()); in TEST()
598 Status UpdateFunction(const NodeDef* function_node) { in UpdateFunction() argument599 auto it = fun_to_grappler_function_item_.find(function_node->op()); in UpdateFunction()602 function_node->op(), in UpdateFunction()622 const TensorId input_tensor = ParseTensorName(function_node->input(i)); in UpdateFunction()658 auto* ctx = GetNodeContext(function_node); in UpdateFunction()661 const string& input = function_node->input(i); in UpdateFunction()712 function_node->name()); in UpdateFunction()1035 Status AddFunction(const NodeDef* function_node) { in AddFunction() argument1036 auto it = fun_to_grappler_function_item_.find(function_node->op()); in AddFunction()1042 CHECK_NOTNULL(function_library_.Find(function_node->op())); in AddFunction()[all …]