Searched refs:getOpOperand (Results 1 – 8 of 8) sorted by relevance
149 op->getOpOperand(i).get().getDefiningOp()); in RewriteOp()158 auto argument_type = op->getOpOperand(1 - i) in RewriteOp()182 op, [&]() { op->getOpOperand(i).set(broadcast.input()); }); in RewriteOp()
117 embedding_op->getOpOperand(embedding_op->getNumOperands() - 1); in UpdateEmbeddingEnqueueOpInput()
328 OpOperand& operand = op->getOpOperand(idx); in MoveTransposeAfter()381 OpOperand& operand = op->getOpOperand(idx); in MoveTransposeAfter()
232 Value getOperand(unsigned idx) { return getOpOperand(idx).get(); } in getOperand()234 return getOpOperand(idx).set(value); in setOperand()261 OpOperand &getOpOperand(unsigned idx) { return getOpOperands()[idx]; } in getOpOperand() function
175 mlir::cast<mlir::ConstantOp>(it->getOpOperand(15).get().getDefiningOp()) in TEST_F()
154 OpOperand &getOpOperand(unsigned idx) { return getOpOperands()[idx]; }155 const OpOperand &getOpOperand(unsigned idx) const {
586 auto &operand = blockIter->getOpOperand(it.value().second); in merge()
523 OpOperand &use = consumer->getOpOperand(consumerIdx); in fuseProducerOfTensor()