Searched refs:func_body_node (Results 1 – 2 of 2) sorted by relevance
1058 for (NodeDef& func_body_node : *item.mutable_function_body().mutable_node()) { in InlineDirectFunctionCall()1059 const string& node_name = func_body_node.name(); in InlineDirectFunctionCall()1062 if (IsIdentity(func_body_node) && output_nodes.count(node_name)) { in InlineDirectFunctionCall()1063 output_tensors.emplace(node_name, func_body_node.input(0)); in InlineDirectFunctionCall()1070 CHECK_EQ(0, func_body_node.input_size()); in InlineDirectFunctionCall()1071 func_body_node.set_op("Identity"); in InlineDirectFunctionCall()1072 (*func_body_node.mutable_attr())["T"] = func_body_node.attr().at("dtype"); in InlineDirectFunctionCall()1073 func_body_node.mutable_attr()->erase("dtype"); in InlineDirectFunctionCall()1074 func_body_node.mutable_attr()->erase("shape"); in InlineDirectFunctionCall()1075 func_body_node.add_input(strings::StrCat(func_inputs->name(), ":", in InlineDirectFunctionCall()[all …]
458 for (const NodeDef& func_body_node : func.node_def()) { in InstantiationBodyParameters() local459 for (auto& attr : func_body_node.attr()) { in InstantiationBodyParameters()672 for (const NodeDef& func_body_node : item.function_body().node()) { in RegisterGrapplerFunctionConnectivity() local674 RegisterFunctionBodyOutputs(flib, func_body_node, connectivity)); in RegisterGrapplerFunctionConnectivity()803 for (const NodeDef& func_body_node : item.function_body().node()) { in MakeFunctionDef() local804 if (!IsIdentity(func_body_node)) continue; in MakeFunctionDef()806 const string& node_name = func_body_node.name(); in MakeFunctionDef()811 if (func_body_node.input_size() == 1) { in MakeFunctionDef()813 << func_body_node.input(0); in MakeFunctionDef()814 output_tensors.emplace(node_name, func_body_node.input(0)); in MakeFunctionDef()