Home
last modified time | relevance | path

Searched refs:old_to_new_output_mapping (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc99 llvm::SmallVector<int, 8> old_to_new_output_mapping; member
301 infos.old_to_new_output_mapping.resize(execute_launch.getNumResults()); in BuildVariableAccessInfo()
304 infos.old_to_new_output_mapping[i] = -1; in BuildVariableAccessInfo()
306 infos.old_to_new_output_mapping[i] = new_output_index; in BuildVariableAccessInfo()
379 for (int i = 0, end = infos.old_to_new_output_mapping.size(); i < end; ++i) { in ReplaceParallelExecute()
380 if (infos.old_to_new_output_mapping[i] < 0) continue; in ReplaceParallelExecute()
382 infos.old_to_new_output_mapping[i] + num_results_before_region)); in ReplaceParallelExecute()
411 for (int i = 0, end = infos.old_to_new_output_mapping.size(); i < end; ++i) { in ReplaceExecute()
412 if (infos.old_to_new_output_mapping[i] < 0) continue; in ReplaceExecute()
414 merged_execute_launch.getResult(infos.old_to_new_output_mapping[i])); in ReplaceExecute()
[all …]