Home
last modified time | relevance | path

Searched refs:first_function (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfusion_utils.cc116 void RenameFunctionNodes(const FunctionDef& first_function, in RenameFunctionNodes() argument
120 GetUniqueNames(first_function.node_def(), *nodes_to_fuse); in RenameFunctionNodes()
298 void MergeNodes(const FunctionDef& first_function, in MergeNodes() argument
302 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in MergeNodes()
377 void LazyConjunctionNodes(const FunctionDef& first_function, in LazyConjunctionNodes() argument
381 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in LazyConjunctionNodes()
384 if_builder.Input(GetOutputNode(first_function, 0), 0, DT_BOOL); in LazyConjunctionNodes()
387 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes()
402 CreateFalsePredicate(first_function.signature().input_arg(), library); in LazyConjunctionNodes()
428 const FunctionDef& first_function, const FunctionDef& second_function, in FuseFunctions() argument
[all …]
H A Dfusion_utils.h59 const FunctionDef& first_function, const FunctionDef& second_function,
62 void MergeNodes(const FunctionDef& first_function,
114 void LazyConjunctionNodes(const FunctionDef& first_function,
126 const FunctionDef& first_function, const FunctionDef& second_function,
H A Dsplit_utils_test.cc207 FunctionDef first = results.first_function; in TEST()
249 FunctionDef first = results.first_function; in TEST()
280 FunctionDef first = results.first_function; in TEST()
315 FunctionDef first = results.first_function; in TEST()
353 FunctionDef first = results.first_function; in TEST()
386 FunctionDef first = results.first_function; in TEST()
415 FunctionDef first = results.first_function; in TEST()
H A Dsplit_utils.cc82 FunctionDef* first_function, FunctionDef* second_function, in InputRewriter() argument
89 first_function_(first_function), in InputRewriter()
432 InitializeSignatures(function, &results.first_function, in SplitFunction()
440 results.first_function.mutable_attr()->insert( in SplitFunction()
450 &results.first_function, in SplitFunction()
476 *results.first_function.add_node_def() = orig_node_def; in SplitFunction()
H A Dsplit_utils.h31 FunctionDef first_function; member
H A Dmake_deterministic.cc326 name_attr_list->set_name(split_results.first_function.signature().name()); in SplitMap()
379 split_results.first_function; in SplitMap()
/aosp_15_r20/external/angle/third_party/spirv-tools/src/test/fuzz/
H A Dshrinker_test.cpp213 bool first_function = true; in TEST() local
215 if (first_function) { in TEST()
216 first_function = false; in TEST()
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/test/fuzz/
Dshrinker_test.cpp213 bool first_function = true; in TEST() local
215 if (first_function) { in TEST()
216 first_function = false; in TEST()
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/test/fuzz/
H A Dshrinker_test.cpp213 bool first_function = true; in TEST() local
215 if (first_function) { in TEST()
216 first_function = false; in TEST()