Home
last modified time | relevance | path

Searched refs:nodes_in_first_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dsplit_utils.cc361 const absl::flat_hash_set<absl::string_view>& nodes_in_first_function, in InitializeSignatures() argument
393 if (nodes_in_first_function.contains(it.second)) { in InitializeSignatures()
405 const absl::flat_hash_set<absl::string_view>& nodes_in_first_function, in SplitFunction() argument
433 &results.second_function, nodes_in_first_function, in SplitFunction()
447 nodes_in_first_function, in SplitFunction()
455 if (!nodes_in_first_function.contains(orig_node_def.name())) { in SplitFunction()
480 !nodes_in_first_function.contains(components[0])) { in SplitFunction()
H A Dsplit_utils.h69 const absl::flat_hash_set<absl::string_view>& nodes_in_first_function,