Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils.cc82 FunctionDef* first_function, FunctionDef* second_function, in InputRewriter() argument
90 second_function_(second_function), in InputRewriter()
433 &results.second_function, nodes_in_first_function, in SplitFunction()
442 results.second_function.mutable_attr()->insert( in SplitFunction()
451 &results.second_function, in SplitFunction()
457 NodeDef& new_node_def = *results.second_function.add_node_def(); in SplitFunction()
498 string& new_ret = (*results.second_function.mutable_ret())[arg_def.name()]; in SplitFunction()
506 *results.second_function.mutable_signature()->add_input_arg() = in SplitFunction()
H A Dfusion_utils.cc299 const FunctionDef& second_function, FunctionDef* fused_function, in MergeNodes() argument
304 fused_function->mutable_node_def()->MergeFrom(second_function.node_def()); in MergeNodes()
378 const FunctionDef& second_function, in LazyConjunctionNodes() argument
398 then_branch.set_name(second_function.signature().name()); in LazyConjunctionNodes()
428 const FunctionDef& first_function, const FunctionDef& second_function, in FuseFunctions() argument
441 if (has_unknown_attrs(first_function) || has_unknown_attrs(second_function)) { in FuseFunctions()
447 FunctionDef setup_function = second_function; in FuseFunctions()
492 get_construction_context(second_function); in FuseFunctions()
H A Dfusion_utils.h59 const FunctionDef& first_function, const FunctionDef& second_function,
63 const FunctionDef& second_function, FunctionDef* fused_function,
115 const FunctionDef& second_function,
126 const FunctionDef& first_function, const FunctionDef& second_function,
H A Dsplit_utils_test.cc208 FunctionDef second = results.second_function; in TEST()
250 FunctionDef second = results.second_function; in TEST()
281 FunctionDef second = results.second_function; in TEST()
316 FunctionDef second = results.second_function; in TEST()
354 FunctionDef second = results.second_function; in TEST()
387 FunctionDef second = results.second_function; in TEST()
416 FunctionDef second = results.second_function; in TEST()
H A Dsplit_utils.h32 FunctionDef second_function; member
H A Dmake_deterministic.cc361 name_attr_list->set_name(split_results.second_function.signature().name()); in SplitMap()
381 split_results.second_function; in SplitMap()