Searched refs:lifted_arg_index_to_oc_cluster_name (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1907 absl::flat_hash_map<int, string>* lifted_arg_index_to_oc_cluster_name, in MakeIdentityNodesForArgsToLift() argument
1920 (*lifted_arg_index_to_oc_cluster_name)[arg_index] = node_name; in MakeIdentityNodesForArgsToLift()
1944 const absl::flat_hash_map<int, string>& lifted_arg_index_to_oc_cluster_name, in RemoveArgsToLiftFromFunctionBody() argument
1974 TF_RET_CHECK(lifted_arg_index_to_oc_cluster_name.contains(i)); in RemoveArgsToLiftFromFunctionBody()
1976 lifted_arg_index_to_oc_cluster_name.at(i)); in RemoveArgsToLiftFromFunctionBody()
2145 absl::flat_hash_map<int, string> lifted_arg_index_to_oc_cluster_name; in LiftOutsideCompilationOnlyArgsFromWhileNode() local
2148 &lifted_arg_index_to_oc_cluster_name, lifted_arg_count)); in LiftOutsideCompilationOnlyArgsFromWhileNode()
2154 args_to_lift, dtypes, lifted_arg_index_to_oc_cluster_name, index_mapping, in LiftOutsideCompilationOnlyArgsFromWhileNode()
2170 args_to_lift, dtypes, lifted_arg_index_to_oc_cluster_name, index_mapping, in LiftOutsideCompilationOnlyArgsFromWhileNode()
2218 absl::flat_hash_map<int, string> lifted_arg_index_to_oc_cluster_name; in LiftOutsideCompilationOnlyArgsFromIfNode() local
[all …]
|