Searched defs:index_mapping (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | rearrange_function_argument.cc | 42 const std::vector<int>& index_mapping) { in ShuffleInputDataTypeAttribute()
57 std::vector<int>* index_mapping) { in InputTypesNeedsRearrange()
109 const std::vector<int>& index_mapping) { in ReorderInputEdges()
134 const std::vector<int>& index_mapping) { in ReorderOutputEdges()
164 const std::vector<int>& index_mapping) { in RearrangeArgNodes()
213 const std::map<int, int>& index_mapping) { in ShuffleOutputDataTypeAttribute() argument
285 std::vector<int> index_mapping; in MaybeRewriteWhileNode() local
392 std::vector<int> index_mapping; in MaybeRewriteIfNode() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | encapsulate_tpu_computations_pass.cc | 1945 const absl::flat_hash_map<int, int>& index_mapping, in RemoveArgsToLiftFromFunctionBody()
1997 Status CleanUpInEdges(const absl::flat_hash_map<int, int>& index_mapping, in CleanUpInEdges() argument
2024 Status UpdateTypeAttribute(const absl::flat_hash_map<int, int>& index_mapping, in UpdateTypeAttribute() argument
2093 absl::flat_hash_map<int, int> index_mapping; in ArgIndexMapping() local
2107 const absl::flat_hash_map<int, int>& index_mapping, in CleanUpRetvalsForWhileBody() argument
2140 absl::flat_hash_map<int, int> index_mapping = in LiftOutsideCompilationOnlyArgsFromWhileNode() local
2207 absl::flat_hash_map<int, int> index_mapping; in LiftOutsideCompilationOnlyArgsFromIfNode() local
2299 absl::flat_hash_map<int, int> index_mapping = in LiftOutsideCompilationOnlyArgsFromCallNode() local
|
| D | distributed_tpu_rewrite_pass.cc | 2812 std::vector<NodeOut> index_mapping; in CreateOrGetPerHostVariableCopy() local
|