Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dresource_op_lifting.cc493 auto old_return = front.getTerminator(); in AppendResourceStoreValueToReturn() local
494 assert(old_return->getNumOperands() == op_->getNumResults()); in AppendResourceStoreValueToReturn()
495 auto new_return_operands = llvm::to_vector<4>(old_return->getOperands()); in AppendResourceStoreValueToReturn()
519 old_return->setOperands(new_return_operands); in AppendResourceStoreValueToReturn()
1017 auto old_return = branch.front().getTerminator(); in HandleCaseOrIfOp() local
1018 OpBuilder builder(old_return); in HandleCaseOrIfOp()
1020 builder.create<ReturnOp>(old_return->getLoc(), new_retvals); in HandleCaseOrIfOp()
1021 old_return->erase(); in HandleCaseOrIfOp()
1143 auto old_return = callee.front().getTerminator(); in HandlePartitionedCallOpCallee() local
1148 old_and_new_retvals.push_back(old_return->getOperand(retval_index)); in HandlePartitionedCallOpCallee()
[all …]