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.cc129 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()
Dmap_vectorization.cc348 auto function_node = branch_func->add_node_def(); in AddBranch() local
349 *function_node = *node; in AddBranch()
351 function_node); in AddBranch()
352 function_node->clear_input(); in AddBranch()
353 function_node->add_input(prev_node_output); in AddBranch()
369 string input_arg_name = strings::StrCat(function_node->name(), "_", i); in AddBranch()
372 function_node->add_input(input_arg_name); in AddBranch()
379 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/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.cc603 Status UpdateFunction(const NodeDef* function_node) { in UpdateFunction() argument
604 auto it = fun_to_grappler_function_item_.find(function_node->op()); in UpdateFunction()
607 function_node->op(), in UpdateFunction()
615 << function_node->op(); in UpdateFunction()
617 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
620 TF_RETURN_IF_ERROR(SetUnknownShape(function_node, i)); in UpdateFunction()
636 const TensorId input_tensor = ParseTensorName(function_node->input(i)); in UpdateFunction()
679 auto* ctx = GetNodeContext(function_node); in UpdateFunction()
682 const string& input = function_node->input(i); in UpdateFunction()
738 function_node->name()); in UpdateFunction()
[all …]
/external/v8/src/compiler/
Dwasm-compiler.cc6946 Node* function_node = Param(param_count + 1); in BuildCapiCallWrapper() local
6948 MachineType::AnyTagged(), function_node, in BuildCapiCallWrapper()