Searched refs:func_node (Results 1 – 4 of 4) sorted by relevance
78 bool IsDirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) { in IsDirectFunctionCall() argument79 return func_node.op() == func.signature().name(); in IsDirectFunctionCall()84 bool IsIndirectFunctionCall(const FunctionDef& func, const NodeDef& func_node) { in IsIndirectFunctionCall() argument85 if (!IsPartitionedCall(func_node) && !IsStatefulPartitionedCall(func_node)) { in IsIndirectFunctionCall()89 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in IsIndirectFunctionCall()95 const NodeDef& func_node) { in FunctionInstantiationAttributes() argument96 if (IsDirectFunctionCall(func, func_node)) { in FunctionInstantiationAttributes()97 return AttrSlice(func_node); in FunctionInstantiationAttributes()99 } else if (IsIndirectFunctionCall(func, func_node)) { in FunctionInstantiationAttributes()100 auto* func_attr = AttrSlice(func_node).Find(kFuncAttr); in FunctionInstantiationAttributes()[all …]
72 for (const NodeDef& func_node : function.node_def()) { in FunctionFindNodeDef() local73 if (func_node.op() != op) { in FunctionFindNodeDef()76 if (func_node.name() + match != func) { in FunctionFindNodeDef()79 return &func_node; in FunctionFindNodeDef()160 const NodeDef* func_node = in OptimizeAndCollectStats() local162 while (func_node != nullptr) { in OptimizeAndCollectStats()163 node_name = func_node->input(0); in OptimizeAndCollectStats()164 func_node = in OptimizeAndCollectStats()167 func_node = FunctionFindNodeDef(function, "StridedSlice", node_name, in OptimizeAndCollectStats()170 if (func_node != nullptr) { in OptimizeAndCollectStats()[all …]
593 for (const NodeDef& func_node : item.function_body().node()) { in MakeFunctionDef() local597 if (IsArg(func_node) || IsRetval(func_node) || in MakeFunctionDef()598 helper.IsInputNode(func_node) || helper.IsOutputNode(func_node)) in MakeFunctionDef()602 *func_def_node = func_node; in MakeFunctionDef()
832 vtn_foreach_cf_node(func_node, &b->functions) { in vtn_build_cfg()833 struct vtn_function *func = vtn_cf_node_as_function(func_node); in vtn_build_cfg()