Searched refs:first_function (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | fusion_utils.cc | 115 void RenameFunctionNodes(const FunctionDef& first_function, in RenameFunctionNodes() argument
119 GetUniqueNames(first_function.node_def(), *nodes_to_fuse); in RenameFunctionNodes()
297 void MergeNodes(const FunctionDef& first_function, in MergeNodes() argument
301 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in MergeNodes()
376 void LazyConjunctionNodes(const FunctionDef& first_function, in LazyConjunctionNodes() argument
380 fused_function->mutable_node_def()->CopyFrom(first_function.node_def()); in LazyConjunctionNodes()
383 if_builder.Input(GetOutputNode(first_function, 0), 0, DT_BOOL); in LazyConjunctionNodes()
386 for (const auto& input_arg : first_function.signature().input_arg()) { in LazyConjunctionNodes()
401 CreateFalsePredicate(first_function.signature().input_arg(), library); in LazyConjunctionNodes()
427 const FunctionDef& first_function, const FunctionDef& second_function, in FuseFunctions() argument
[all …]
|
| D | fusion_utils.h | 59 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,
|