Searched refs:new_while_operands (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tensor_list_ops_decomposition.cc | 189 auto new_while_operands = llvm::to_vector<8>(while_op.getOperands()); in HandleWhileOp() local
193 new_while_operands.push_back(it->getSecond().size); in HandleWhileOp()
196 while_op.getLoc(), body.getFunctionType().getInputs(), new_while_operands, in HandleWhileOp()
311 auto new_while_operands = llvm::to_vector<8>(while_op.getOperands()); in HandleWhileRegionOp() local
315 new_while_operands.push_back(it->getSecond().size); in HandleWhileRegionOp()
319 new_while_operands, while_op->getAttrs()); in HandleWhileRegionOp()
|
| H A D | stack_ops_decomposition.cc | 170 auto new_while_operands = llvm::to_vector<8>(while_op.getOperands()); in HandleWhileOp() local
176 new_while_operands.push_back(it->getSecond()); in HandleWhileOp()
179 while_op.getLoc(), body.getFunctionType().getInputs(), new_while_operands, in HandleWhileOp()
|
| H A D | resource_op_lifting.cc | 555 SmallVector<Value, 4> new_while_operands; in HoistResourcesOutOfWhileRegion() local
557 new_while_operands.resize(num_extra_results); in HoistResourcesOutOfWhileRegion()
563 new_while_operands[index] = it.second.hoisted_read; in HoistResourcesOutOfWhileRegion()
565 op.getOperation()->insertOperands(op.getNumOperands(), new_while_operands); in HoistResourcesOutOfWhileRegion()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_ops_n_z.cc | 3230 SmallVector<Value, 4> new_while_operands; in matchAndRewrite() local
3232 new_while_operands.reserve(new_num_operands); in matchAndRewrite()
3238 new_while_operands.push_back(while_op.getOperand(op_idx)); in matchAndRewrite()
3244 while_op.getLoc(), new_result_types, new_while_operands, in matchAndRewrite()
|