Searched refs:InstantiateAssociatedFunction (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1682 xla::StatusOr<std::unique_ptr<FunctionBody>> InstantiateAssociatedFunction( in InstantiateAssociatedFunction() function
1717 InstantiateAssociatedFunction(if_node, "then_branch", fld)); in FindArgsToLiftForIfNode()
1721 InstantiateAssociatedFunction(if_node, "else_branch", fld)); in FindArgsToLiftForIfNode()
2152 InstantiateAssociatedFunction(*while_node, "cond", fld)); in LiftOutsideCompilationOnlyArgsFromWhileNode()
2167 InstantiateAssociatedFunction(*while_node, "body", fld)); in LiftOutsideCompilationOnlyArgsFromWhileNode()
2225 InstantiateAssociatedFunction(*if_node, "then_branch", fld)); in LiftOutsideCompilationOnlyArgsFromIfNode()
2240 InstantiateAssociatedFunction(*if_node, "else_branch", fld)); in LiftOutsideCompilationOnlyArgsFromIfNode()
2366 InstantiateAssociatedFunction(*n, "body", fld)); in LiftOutsideCompilationOnlyArgs()
2377 InstantiateAssociatedFunction(*n, "then_branch", fld)); in LiftOutsideCompilationOnlyArgs()
2386 InstantiateAssociatedFunction(*n, "else_branch", fld)); in LiftOutsideCompilationOnlyArgs()
[all …]
|