Lines Matching refs:tensorExps

137   unsigned e = tensorExps.size();  in addExp()
138 tensorExps.push_back(TensorExp(k, e0, e1, v, op)); in addExp()
233 if (tensorExps[e].kind == kTensor && tensorExps[e].tensor == outTensor) in optimizeSet()
303 switch (tensorExps[e].kind) { in isSingleCondition()
306 return tensorExps[e].tensor == t; in isSingleCondition()
341 return isSingleCondition(t, tensorExps[e].children.e0); in isSingleCondition()
350 assert(!maybeZero(tensorExps[e].children.e1)); in isSingleCondition()
351 return isSingleCondition(t, tensorExps[e].children.e0); in isSingleCondition()
355 assert(isInvariant(tensorExps[e].children.e1)); in isSingleCondition()
356 return isSingleCondition(t, tensorExps[e].children.e0); in isSingleCondition()
361 if (isSingleCondition(t, tensorExps[e].children.e0)) in isSingleCondition()
362 return isSingleCondition(t, tensorExps[e].children.e1) || in isSingleCondition()
363 isInvariant(tensorExps[e].children.e1); in isSingleCondition()
364 if (isSingleCondition(t, tensorExps[e].children.e1)) in isSingleCondition()
365 return isInvariant(tensorExps[e].children.e0); in isSingleCondition()
370 return isSingleCondition(t, tensorExps[e].children.e0) && in isSingleCondition()
371 isSingleCondition(t, tensorExps[e].children.e1); in isSingleCondition()
482 switch (tensorExps[e].kind) { in dumpExp()
485 if (tensorExps[e].tensor == syntheticTensor) in dumpExp()
487 else if (tensorExps[e].tensor == outTensor) in dumpExp()
489 llvm::dbgs() << "tensor_" << tensorExps[e].tensor; in dumpExp()
495 llvm::dbgs() << "index_" << tensorExps[e].index; in dumpExp()
530 llvm::dbgs() << kindToOpSymbol(tensorExps[e].kind) << " "; in dumpExp()
531 dumpExp(tensorExps[e].children.e0); in dumpExp()
555 dumpExp(tensorExps[e].children.e0); in dumpExp()
556 llvm::dbgs() << " " << kindToOpSymbol(tensorExps[e].kind) << " "; in dumpExp()
557 dumpExp(tensorExps[e].children.e1); in dumpExp()
612 Kind kind = tensorExps[e].kind; in buildLattices()
626 t = tensorExps[e].tensor; in buildLattices()
670 return mapSet(kind, buildLattices(tensorExps[e].children.e0, i), in buildLattices()
671 tensorExps[e].val); in buildLattices()
675 return mapSet(kind, buildLattices(tensorExps[e].children.e0, i), Value(), in buildLattices()
676 tensorExps[e].op); in buildLattices()
684 unsigned child0 = buildLattices(tensorExps[e].children.e0, i); in buildLattices()
685 UnaryOp unop = cast<UnaryOp>(tensorExps[e].op); in buildLattices()
714 buildLattices(tensorExps[e].children.e0, i), in buildLattices()
715 buildLattices(tensorExps[e].children.e1, i)); in buildLattices()
733 assert(!maybeZero(tensorExps[e].children.e1)); in buildLattices()
735 buildLattices(tensorExps[e].children.e0, i), in buildLattices()
736 buildLattices(tensorExps[e].children.e1, i)); in buildLattices()
753 buildLattices(tensorExps[e].children.e0, i), in buildLattices()
754 buildLattices(tensorExps[e].children.e1, i)); in buildLattices()
761 assert(isInvariant(tensorExps[e].children.e1)); in buildLattices()
763 buildLattices(tensorExps[e].children.e0, i), in buildLattices()
764 buildLattices(tensorExps[e].children.e1, i)); in buildLattices()
773 unsigned child0 = buildLattices(tensorExps[e].children.e0, i); in buildLattices()
774 unsigned child1 = buildLattices(tensorExps[e].children.e1, i); in buildLattices()
775 BinaryOp binop = cast<BinaryOp>(tensorExps[e].op); in buildLattices()
809 if (tensorExps[e].kind == kInvariant) { in maybeZero()
810 if (auto c = tensorExps[e].val.getDefiningOp<complex::ConstantOp>()) { in maybeZero()
815 if (auto c = tensorExps[e].val.getDefiningOp<arith::ConstantIntOp>()) in maybeZero()
817 if (auto c = tensorExps[e].val.getDefiningOp<arith::ConstantFloatOp>()) in maybeZero()
824 return tensorExps[e].kind == kInvariant; in isInvariant()
829 Type dtp = tensorExps[e].val.getType(); in inferType()
1071 switch (tensorExps[e].kind) { in buildExp()
1192 *tensorExps[e].op->getBlock()->getParent(), {v0}); in buildExp()
1194 return buildUnaryPresent(rewriter, loc, tensorExps[e].op, v0); in buildExp()
1196 return buildBinaryOverlap(rewriter, loc, tensorExps[e].op, v0, v1); in buildExp()