Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
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()
Dmap_vectorization.cc368 auto function_node = branch_func->add_node_def(); in AddBranch() local
369 *function_node = *node; in AddBranch()
371 function_node); in AddBranch()
372 function_node->clear_input(); in AddBranch()
373 function_node->add_input(prev_node_output); in AddBranch()
389 string input_arg_name = strings::StrCat(function_node->name(), "_", i); in AddBranch()
392 function_node->add_input(input_arg_name); in AddBranch()
399 prev_node_output = strings::StrCat(function_node->name(), ":handle:0"); in AddBranch()
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()
/external/tensorflow/tensorflow/tools/graph_transforms/
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()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc106 Node* function_node = g.AddNode(std::move(function_node_def), &status); in ConstructFunctionGraph() local
109 g.AddEdge(input_nodes[i], 0, function_node, i); in ConstructFunctionGraph()
140 g.AddEdge(function_node, i, output_node, 0); in ConstructFunctionGraph()
162 TF_RETURN_IF_ERROR(InlineFunctionBody(pruned_flib_def, &g, function_node, in ConstructFunctionGraph()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc756 Status UpdateFunction(const NodeDef* function_node) { in UpdateFunction() argument
758 TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(*function_node, &function)); in UpdateFunction()
772 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
775 TF_RETURN_IF_ERROR(SetUnknownShape(function_node, i)); in UpdateFunction()
792 const TensorId input_tensor = ParseTensorName(function_node->input(i)); in UpdateFunction()
846 << "A function node (" << function_node->name() in UpdateFunction()
849 << "function_node: " << function_node->ShortDebugString() << "\n" in UpdateFunction()
868 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
871 const string& input = function_node->input(i); in UpdateFunction()
928 function_node->name()); in UpdateFunction()
[all …]