Home
last modified time | relevance | path

Searched refs:OpTrait (Results 1 – 25 of 166) sorted by relevance

1234567

/external/llvm-project/mlir/include/mlir/TableGen/
DOpTrait.h32 class OpTrait {
46 explicit OpTrait(Kind kind, const llvm::Record *def);
49 static OpTrait create(const llvm::Init *init);
63 class NativeOpTrait : public OpTrait {
68 static bool classof(const OpTrait *t) { return t->getKind() == Kind::Native; } in classof()
72 class PredOpTrait : public OpTrait {
80 static bool classof(const OpTrait *t) { return t->getKind() == Kind::Pred; } in classof()
84 class InternalOpTrait : public OpTrait {
89 static bool classof(const OpTrait *t) { in classof()
95 class InterfaceOpTrait : public OpTrait {
[all …]
DOperator.h180 const OpTrait *getTrait(llvm::StringRef trait) const;
211 using const_trait_iterator = const OpTrait *;
324 SmallVector<OpTrait, 4> traits;
DSideEffects.h52 static bool classof(const OpTrait *t);
/external/llvm-project/mlir/lib/TableGen/
DOpTrait.cpp24 OpTrait OpTrait::create(const llvm::Init *init) { in create()
27 return OpTrait(Kind::Pred, def); in create()
29 return OpTrait(Kind::Internal, def); in create()
31 return OpTrait(Kind::Interface, def); in create()
33 return OpTrait(Kind::Native, def); in create()
36 OpTrait::OpTrait(Kind kind, const llvm::Record *def) : def(def), kind(kind) {} in OpTrait() function in OpTrait
/external/llvm-project/mlir/lib/IR/
DOperation.cpp119 needsOperandStorage = !abstractOp->hasTrait<OpTrait::ZeroOperands>(); in create()
697 OpFoldResult OpTrait::impl::foldIdempotent(Operation *op) { in foldIdempotent()
707 OpFoldResult OpTrait::impl::foldInvolution(Operation *op) { in foldInvolution()
717 LogicalResult OpTrait::impl::verifyZeroOperands(Operation *op) { in verifyZeroOperands()
723 LogicalResult OpTrait::impl::verifyOneOperand(Operation *op) { in verifyOneOperand()
729 LogicalResult OpTrait::impl::verifyNOperands(Operation *op, in verifyNOperands()
738 LogicalResult OpTrait::impl::verifyAtLeastNOperands(Operation *op, in verifyAtLeastNOperands()
758 LogicalResult OpTrait::impl::verifyIsIdempotent(Operation *op) { in verifyIsIdempotent()
766 LogicalResult OpTrait::impl::verifyIsInvolution(Operation *op) { in verifyIsInvolution()
775 OpTrait::impl::verifyOperandsAreSignlessIntegerLike(Operation *op) { in verifyOperandsAreSignlessIntegerLike()
[all …]
DSymbolTable.cpp63 if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in collectValidReferencesFor()
96 if (!op.hasTrait<OpTrait::SymbolTable>()) { in walkSymbolTable()
112 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>() && in SymbolTable()
236 while (!from->hasTrait<OpTrait::SymbolTable>()) { in getNearestSymbolTable()
254 bool isSymbolTable = op->hasTrait<OpTrait::SymbolTable>(); in walkSymbolTables()
281 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>()); in lookupSymbolIn()
305 assert(symbolTableOp->hasTrait<OpTrait::SymbolTable>()); in lookupSymbolInImpl()
319 if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in lookupSymbolInImpl()
326 if (!symbolTableOp || !symbolTableOp->hasTrait<OpTrait::SymbolTable>()) in lookupSymbolInImpl()
532 if (!from->hasTrait<OpTrait::SymbolTable>()) in walkSymbolUses()
[all …]
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgTraits.h21 namespace OpTrait {
32 class Impl : public OpTrait::TraitBase<ConcreteType, NInputs<N>::Impl> {
57 class Impl : public OpTrait::TraitBase<ConcreteType, NOutputs<N>::Impl> {
72 : public OpTrait::TraitBase<ConcreteType, StructuredOpTraits> {
77 if (failed(OpTrait::impl::verifyAtLeastNOperands(op, nOperands))) in verifyTrait()
96 : public OpTrait::TraitBase<ConcreteType, NamedStructuredOpTrait> {
114 OpTrait::impl::verifyAtLeastNOperands(op, nInputAndBufferOperands))) in verifyTrait()
/external/llvm-project/mlir/docs/
DTraits.md20 `AttributeTrait::TraitBase`. For operations, this is `OpTrait::TraitBase`. For
45 class MyTrait : public OpTrait::TraitBase<ConcreteType, MyTrait> {
69 class MyTrait : public OpTrait::TraitBase<ConcreteType, MyTrait> {
85 class MyTrait : public OpTrait::TraitBase<ConcreteType, MyTrait> {
194 * `OpTrait::AffineScope` -- `AffineScope`
209 * `OpTrait::AutomaticAllocationScope` -- `AutomaticAllocationScope`
220 * `OpTrait::ResultsBroadcastableShape` -- `ResultsBroadcastableShape`
237 * `OpTrait::IsCommutative` -- `Commutative`
244 * `OpTrait::ElementwiseMappable` -- `ElementwiseMappable`
254 `OpTrait::ElementwiseMappable` for the precise requirements.
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfold_broadcast.cc64 if (op->hasTrait<OpTrait::ResultsBroadcastableShape>()) in matchAndRewrite()
65 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in matchAndRewrite()
94 if (!OpTrait::util::getBroadcastedShape( in RewriteBatchMatMulV2Op()
128 return RewriteOp(op, rewriter, OpTrait::util::getBroadcastedShape); in RewriteEqOp()
/external/llvm-project/mlir/lib/Transforms/
DSymbolDCE.cpp37 if (!symbolTableOp->hasTrait<OpTrait::SymbolTable>()) { in runOnOperation()
61 if (!nestedSymbolTable->hasTrait<OpTrait::SymbolTable>()) in runOnOperation()
106 if (op->hasTrait<OpTrait::SymbolTable>()) { in computeLiveness()
DLoopInvariantCodeMotion.cpp55 if (!op->hasTrait<OpTrait::HasRecursiveSideEffects>()) in canBeHoisted()
61 } else if (!op->hasTrait<OpTrait::HasRecursiveSideEffects>()) { in canBeHoisted()
/external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/
DAffineOps.h84 : public Op<AffineDmaStartOp, OpTrait::MemRefsNormalizable,
85 OpTrait::VariadicOperands, OpTrait::ZeroResult> {
273 : public Op<AffineDmaWaitOp, OpTrait::MemRefsNormalizable,
274 OpTrait::VariadicOperands, OpTrait::ZeroResult> {
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.h150 : public Op<DmaStartOp, OpTrait::VariadicOperands, OpTrait::ZeroResult> {
273 : public Op<DmaWaitOp, OpTrait::VariadicOperands, OpTrait::ZeroResult> {
/external/llvm-project/mlir/include/mlir/Dialect/SPIRV/
DSPIRVOCLOps.td23 class SPV_OCLOp<string mnemonic, int opcode, list<OpTrait> traits = []> :
28 Type operandType, list<OpTrait> traits = []> :
49 list<OpTrait> traits = []> :
54 Type operandType, list<OpTrait> traits = []> :
76 list<OpTrait> traits = []> :
/external/llvm-project/mlir/lib/Dialect/
DTraits.cpp16 bool OpTrait::util::staticallyKnownBroadcastable(ArrayRef<int64_t> shape1, in staticallyKnownBroadcastable()
33 bool OpTrait::util::getBroadcastedShape(ArrayRef<int64_t> shape1, in getBroadcastedShape()
106 Type OpTrait::util::getBroadcastedType(Type type1, Type type2, in getBroadcastedType()
189 LogicalResult OpTrait::impl::verifyCompatibleOperandBroadcast(Operation *op) { in verifyCompatibleOperandBroadcast()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc141 if (inst->hasTrait<mlir::OpTrait::AttrSizedOperandSegments>()) { in GetAttributesToIgnore()
143 llvm::StringRef attr_name = mlir::OpTrait::AttrSizedOperandSegments< in GetAttributesToIgnore()
148 if (inst->hasTrait<mlir::OpTrait::AttrSizedResultSegments>()) { in GetAttributesToIgnore()
150 llvm::StringRef attr_name = mlir::OpTrait::AttrSizedResultSegments< in GetAttributesToIgnore()
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-decl.td13 class NS_Op<string mnemonic, list<OpTrait> traits> :
66 …OpTrait::AtLeastNRegions<1>::Impl, ::mlir::OpTrait::AtLeastNResults<1>::Impl, ::mlir::OpTrait::Zer…
67 // CHECK-NOT: ::mlir::OpTrait::IsIsolatedFromAbove
131 // CHECK: OpTrait::ZeroOperands
138 // CHECK: OpTrait::OneOperand
145 // CHECK: OpTrait::NOperands<2>::Impl
/external/llvm-project/mlir/include/mlir/Dialect/LLVMIR/
DROCDLOps.td33 class ROCDL_Op<string mnemonic, list<OpTrait> traits = []> :
42 list<OpTrait> traits = []> :
51 int parameter, list<OpTrait> traits = []> :
109 class ROCDL_Mfma_IntrOp<string mnemonic, list<OpTrait> traits = []> :
DNVVMOps.td33 class NVVM_Op<string mnemonic, list<OpTrait> traits = []> :
42 list<int> overloadedOperands, list<OpTrait> traits,
53 list<OpTrait> traits = []> :
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_functional_ops.cc62 !op->hasTrait<OpTrait::IsTerminator>()) in IsSideEffectFree()
126 if (op_to_inline.hasTrait<OpTrait::IsTerminator>()) { in matchAndRewrite()
Dpost_quantize.cc139 op->user_begin()->hasTrait<OpTrait::IsTerminator>()) in matchAndRewrite()
159 if (op.getOperation()->template hasTrait<OpTrait::IsTerminator>()) { in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Transforms/
DFoldUtils.h80 typename std::enable_if<OpTy::template hasTrait<OpTrait::OneResult>(),
90 typename std::enable_if<OpTy::template hasTrait<OpTrait::ZeroResult>(),
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinOps.td29 class Builtin_Op<string mnemonic, list<OpTrait> traits = []> :
135 friend class OpTrait::FunctionLike<FuncOp>;
138 /// OpTrait::FunctionLike.
141 /// Returns the number of results. This is a hook for OpTrait::FunctionLike.
144 /// Hook for OpTrait::FunctionLike, called after verifying that the 'type'
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DOpToFuncCallLowering.h47 std::is_base_of<OpTrait::OneResult<SourceOp>, SourceOp>::value, in matchAndRewrite()
50 static_assert(std::is_base_of<OpTrait::SameOperandsAndResultType<SourceOp>, in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Affine/Utils/
DUtils.cpp150 !forOp->getParentOp()->hasTrait<OpTrait::AffineScope>()) in affineParallelize()
224 hoistedIfOp->getParentWithTrait<OpTrait::IsIsolatedFromAbove>(), in hoistAffineIfOp()

1234567