Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_merge_variables_with_execute.cc389 Operation* new_parallel_execute_op = new_parallel_execute.getOperation(); in ReplaceParallelExecute() local
392 new_parallel_execute_op->getResult(i)); in ReplaceParallelExecute()
399 new_parallel_execute_op->getNumResults(); in ReplaceParallelExecute()
402 .replaceAllUsesWith(new_parallel_execute_op->getResult( in ReplaceParallelExecute()
411 old_region_results[i].replaceAllUsesWith(new_parallel_execute_op->getResult( in ReplaceParallelExecute()
428 for (auto region : llvm::zip(new_parallel_execute_op->getRegions(), in ReplaceParallelExecute()