Searched refs:lifted_arg_count (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1878 int* lifted_arg_count, bool* rewritten);
1882 FunctionLibraryDefinition* fld, int* lifted_arg_count, in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef() argument
1886 fbody.graph, flr, fld, lifted_arg_count, rewritten)); in LiftOutsideCompilationOnlyArgsAndReplaceFunctionDef()
1908 int* lifted_arg_count) { in MakeIdentityNodesForArgsToLift() argument
1918 g->NewName(absl::StrCat("lifted_arg", *lifted_arg_count)); in MakeIdentityNodesForArgsToLift()
1919 (*lifted_arg_count)++; in MakeIdentityNodesForArgsToLift()
2125 int* lifted_arg_count, bool* rewritten) { in LiftOutsideCompilationOnlyArgsFromWhileNode() argument
2148 &lifted_arg_index_to_oc_cluster_name, lifted_arg_count)); in LiftOutsideCompilationOnlyArgsFromWhileNode()
2197 int* lifted_arg_count, in LiftOutsideCompilationOnlyArgsFromIfNode() argument
2221 &lifted_arg_index_to_oc_cluster_name, lifted_arg_count)); in LiftOutsideCompilationOnlyArgsFromIfNode()
[all …]
|
| D | encapsulate_tpu_computations_pass.h | 66 const string& outside_compilation_attr_name, int* lifted_arg_count,
|