Lines Matching refs:sparseOut
64 idxs(numTensors, std::vector<Value>(numLoops)), redVal(), sparseOut(op), in CodeGen()
95 OpOperand *sparseOut; member
324 OpOperand **sparseOut, in isAdmissableTensorExp() argument
363 *sparseOut = lhs; in isAdmissableTensorExp()
550 } else if (t == codegen.sparseOut) { in genBuffers()
786 if (t == codegen.sparseOut) in genTensorLoad()
810 if (t == codegen.sparseOut) { in genTensorStore()
1048 OpOperand *lhs = codegen.sparseOut; in genExpansion()
1127 if (codegen.sparseOut && !isReduction) in isVectorFor()
1147 if (codegen.sparseOut) in isParallelFor()
1399 if (codegen.sparseOut && !codegen.expValues) { in genLocals()
1694 codegen.sparseOut == lhs); in genResult()
1746 OpOperand *sparseOut = nullptr; in matchAndRewrite() local
1748 if (!isAdmissableTensorExp(merger, op, topSort, exp, &sparseOut, in matchAndRewrite()
1753 merger.setHasSparseOut(sparseOut != nullptr); in matchAndRewrite()
1754 CodeGen codegen(options, numTensors, numLoops, sparseOut, outerParNest); in matchAndRewrite()