Searched defs:outputTensor (Results 1 – 5 of 5) sorted by relevance
253 makeTransposeOp(OpBuilder & b,Location loc,Value inputTensor,Value outputTensor,ArrayRef<int64_t> transposeVector) makeTransposeOp() argument 761 Value outputTensor = operands[opOperand.getOperandNumber()]; insertSlicesBack() local
639 Value outputTensor = rewriter.create<tensor::ExtractSliceOp>( in buildPackingLoopNestImpl() local
248 Value outputTensor = rewriter.create<tensor::EmptyOp>( in rewriteInIm2Col() local
788 Value outputTensor = opBuilder.create<tensor::EmptyOp>( broadcastDynamicDimension() local 863 Value outputTensor = rewriter.create<tensor::EmptyOp>( emitElementwiseComputation() local
1897 TensorType outputTensor = llvm::dyn_cast<TensorType>(outputs[0]); areCastCompatible() local