Searched refs:inputTensor (Results 1 – 5 of 5) sorted by relevance
228 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor() local230 getProducerOfTensor(inputTensor, producerOpResult); in fuseProducerOfTensor()249 Value inputTensor = consumerOpOperand.get(); in fuseProducerOfTensor() local252 auto sliceOp = inputTensor.getDefiningOp<tensor::ExtractSliceOp>(); in fuseProducerOfTensor()255 << "\nNot fusable, not an extract_slice op: " << inputTensor); in fuseProducerOfTensor()
254 auto inputTensor = recip.getInput1(); in matchAndRewrite() local258 notifyIfNotConstantFloatTosaTensor(inputTensor, recip, rewriter); in matchAndRewrite()265 matchPattern(inputTensor, m_Constant(&inputValues)); in matchAndRewrite()
252 makeTransposeOp(OpBuilder & b,Location loc,Value inputTensor,Value outputTensor,ArrayRef<int64_t> transposeVector) makeTransposeOp() argument
1915 if (auto inputTensor = llvm::dyn_cast<RankedTensorType>(inputs[0])) { in build() 1916 if (!llvm::isa<IndexType>(inputTensor.getElementType()) || in build() 1917 inputTensor.getRank() != 1) in build() 1889 if (auto inputTensor = llvm::dyn_cast<RankedTensorType>(inputs[0])) { areCastCompatible() local