Home
last modified time | relevance | path

Searched refs:function_node (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfusion_utils.cc130 for (NodeDef& function_node : *nodes_to_fuse) { in RenameFunctionNodes()
132 gtl::FindOrNull(changed_node_names, function_node.name())) { in RenameFunctionNodes()
133 function_node.set_name(*new_name); in RenameFunctionNodes()
136 for (string& input : *function_node.mutable_input()) { in RenameFunctionNodes()
195 for (NodeDef& function_node : *nodes_to_fuse) { in GetUniqueSignature()
196 for (auto& node_input : *function_node.mutable_input()) { in GetUniqueSignature()
216 for (NodeDef& function_node : *nodes_to_fuse) { in FuseFunctionNodes()
217 for (auto& node_input : *function_node.mutable_input()) { in FuseFunctionNodes()
H A Dfunction_utils_test.cc127 for (const NodeDef& function_node : function.node_def()) { in TEST() local
128 EXPECT_NE(node.name(), function_node.name()); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dinline_partitionedcall.cc110 for (const NodeDef& function_node : function->node_def()) { in InlinePartitionedCall() local
112 *new_node = function_node; in InlinePartitionedCall()
113 new_node->set_name(absl::StrCat(prefix, "/", function_node.name())); in InlinePartitionedCall()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcluster_function_library_runtime.cc105 TF_ASSIGN_OR_RETURN(Node * function_node, in ConstructFunctionGraph()
108 g.AddEdge(input_nodes[i], 0, function_node, i); in ConstructFunctionGraph()
139 g.AddEdge(function_node, i, output_node, 0); in ConstructFunctionGraph()
161 TF_RETURN_IF_ERROR(InlineFunctionBody(pruned_flib_def, &g, function_node, in ConstructFunctionGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties.cc785 Status UpdateFunction(const NodeDef* function_node) { in UpdateFunction() argument
787 TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(*function_node, &function)); in UpdateFunction()
801 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
804 TF_RETURN_IF_ERROR(SetUnknownShape(function_node, i)); in UpdateFunction()
821 const TensorId input_tensor = ParseTensorName(function_node->input(i)); in UpdateFunction()
875 << "A function node (" << function_node->name() in UpdateFunction()
878 << "function_node: " << function_node->ShortDebugString() << "\n" in UpdateFunction()
897 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
900 const string& input = function_node->input(i); in UpdateFunction()
957 function_node->name()); in UpdateFunction()
[all …]
/aosp_15_r20/external/cronet/build/gn_ast/
H A Dgn_ast.py295 def __init__(self, function_node: dict, name_node: dict):
296 super().__init__(function_node)
338 return Target(function_node=node, name_node=name_node)
/aosp_15_r20/external/angle/build/gn_ast/
H A Dgn_ast.py295 def __init__(self, function_node: dict, name_node: dict):
296 super().__init__(function_node)
338 return Target(function_node=node, name_node=name_node)
/aosp_15_r20/external/icing/icing/query/advanced_query_parser/
H A Dparser.cc128 ICING_ASSIGN_OR_RETURN(std::unique_ptr<FunctionNode> function_node, in ConsumeMember()
133 std::move(function_node)); in ConsumeMember()