Searched refs:transposeOps (Results 1 – 3 of 3) sorted by relevance
399 tensor::PadOp &hoistedOp, SmallVectorImpl<GenericOp> &transposeOps) { in hoistPaddingOnTensors() argument506 transposeOps.push_back( in hoistPaddingOnTensors()508 paddedTensor = transposeOps.back()->getResult(0); in hoistPaddingOnTensors()549 transposeOps.push_back( in hoistPaddingOnTensors()551 newResult = transposeOps.back()->getResult(0); in hoistPaddingOnTensors()
434 SmallVector<GenericOp> transposeOps; in returningMatchAndRewrite() local441 padOp, en.value(), transposeVector, hoistedOp, transposeOps); in returningMatchAndRewrite()447 for (GenericOp transposeOp : transposeOps) in returningMatchAndRewrite()
72 tensor::PadOp &hoistedOp, SmallVectorImpl<GenericOp> &transposeOps);