Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_head_tail_outside_compilation.cc315 Operation* old_terminator = cluster.GetBody().getTerminator(); in UpdateClusterResults() local
316 builder->setInsertionPoint(old_terminator); in UpdateClusterResults()
317 builder->create<tf_device::ReturnOp>(old_terminator->getLoc(), in UpdateClusterResults()
319 old_terminator->erase(); in UpdateClusterResults()
Dtpu_merge_variables_with_execute.cc387 Operation* old_terminator = execute_region->front().getTerminator(); in ReplaceParallelExecute() local
389 builder->create<tf_device::ReturnOp>(old_terminator->getLoc(), in ReplaceParallelExecute()
391 old_terminator->erase(); in ReplaceParallelExecute()
Dtensor_list_ops_decomposition.cc134 auto old_terminator = block.getTerminator(); in AddTensorListSizesToTerminator() local
135 auto new_outputs = llvm::to_vector<8>(old_terminator->getOperands()); in AddTensorListSizesToTerminator()
138 for (auto retval : llvm::enumerate(old_terminator->getOperands())) { in AddTensorListSizesToTerminator()
145 OpBuilder(old_terminator) in AddTensorListSizesToTerminator()
146 .create<TerminatorOp>(old_terminator->getLoc(), new_outputs); in AddTensorListSizesToTerminator()
147 old_terminator->erase(); in AddTensorListSizesToTerminator()